【HDU6667】Roundgod and Milk Tea【貪心】

題目大意:給你ai,bi,限制ai不能流向bi,求最大流node

題解:貪心,對於第i個班級,考慮前i-1個班級匹配完剩餘多少a,b,將這些ab對第i個班級進行貪心匹配ios

匹配完若第i個班級還有剩餘的ab,考慮將前面的已匹配拆分spa

考慮如何拆分code

若存在a1-b2這樣一個匹配,那麼能夠將它拆開,並與aibi匹配成a1-bi,ai-b2,這樣答案就多了1blog

因而能夠想到每新貪心一個班級時,先用cnta和cntb進行貪心,若剩餘ab,則在前面找已匹配進行拆分從新匹配使得答案增長it

時間複雜度o(n)io

代碼:class

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<algorithm>
#define ll long long
using namespace std;
int T,n;
struct node
{
    ll a,b;
}c[1000001];
ll ans,cnta,cntb,tc;
ll ta[1000001],tb[1000001];
int main()
{
    scanf("%d",&T);
    while(T--)
    {
      scanf("%d",&n);
      for(int i=1;i<=n;i++)scanf("%lld%lld",&c[i].a,&c[i].b);
      cnta=cntb=ans=0;int ka=1,kb=1;
      ll tta,ttb;
      for(int i=1;i<=n;i++)
      {
          //printf("%d %lld %lld\n",i,cnta,cntb);
          tta=c[i].a;ttb=c[i].b;
        if(cnta>=c[i].b)cnta-=c[i].b,ans+=c[i].b,c[i].b=0;
        else c[i].b-=cnta,ans+=cnta,cnta=0;
        if(cntb>=c[i].a)cntb-=c[i].a,ans+=c[i].a,c[i].a=0;
        else c[i].a-=cntb,ans+=cntb,cntb=0;
          tc=min(c[i].a,c[i].b);
        while(tc && ka<i)
        {
          if(ta[ka]>=tc)ans+=tc,c[i].a-=tc,c[i].b-=tc,ta[ka]-=tc,tc=0;
          else ans+=ta[ka],c[i].a-=ta[ka],c[i].b-=ta[ka],tc-=ta[ka],ka++;
        }
        while(tc && kb<i)
        {
          if(tb[kb]>=tc)ans+=tc,c[i].a-=tc,c[i].b-=tc,tb[kb]-=tc,tc=0;
          else ans+=tb[kb],c[i].a-=tb[kb],c[i].b-=tb[kb],tc-=tb[kb],kb++;
        }
        ta[i]=tta-c[i].a;tb[i]=ttb-c[i].b;
        cnta+=c[i].a;cntb+=c[i].b;
        //printf("%d %lld %lld %lld\n",i,cnta,cntb,ans);
      }
      //printf("%lld %lld %lld\n",cnta,cntb,ans);
      
      printf("%lld\n",ans);
    }
    return 0;
}

心得:對貪心還不太會,還要繼續努力stream

相關文章
相關標籤/搜索