bzoj 4464 : [Jsoi2013]旅行時的困惑
阿新 • • 發佈:2017-07-07
def else print 反向 註意 using zoj 旅行 上界
網絡流建圖。
從S向每個點連邊,從每個點向T連邊。
每條樹邊反向連一條下界為1,上界inf的邊。
跑最小流。
註意加當前弧優化。
#include<cstdio> #include<algorithm> #define inf 0x3f3f3f3f #define N 500005 #define M 5000005 using namespace std; int head[N],ver[M],nxt[M],f[M],tot,ch[N],cur[M]; void add(int a,int b,int c) { tot++;nxt[tot]=head[a];head[a]=tot;ver[tot]=b;f[tot]=c; tot++;nxt[tot]=head[b];head[b]=tot;ver[tot]=a;f[tot]=0; return ; } int S,T,tim; int vis[100005],q[M]; bool tell() { tim++; int l=1,r=1;q[1]=S; ch[S]=0;vis[S]=tim; while(l<=r) { int tmp=q[l++]; for(int i=head[tmp];i;i=nxt[i]) { if(f[i]&&vis[ver[i]]!=tim) { ch[ver[i]]=ch[tmp]+1; vis[ver[i]]=tim; q[++r]=ver[i]; } } } return vis[T]==tim; } int zeng(int a,int b) { if(a==T)return b; int r=0; for(int i=cur[a];i&&b>r;i=nxt[i]) { if(ch[ver[i]]==ch[a]+1&&f[i]) { int t=zeng(ver[i],min(b-r,f[i])); r+=t;f[i]-=t;f[i^1]+=t; if(f[i]>0)cur[a]=i; } } if(!r)ch[a]=-1; return r; } int dinic() { int r=0,t; while(tell()) { for(int i=0;i<=T;i++)cur[i]=head[i]; while(t=zeng(S,inf))r+=t; } return r; } int n; int ru[N],chu[N]; int main() { scanf("%d",&n);int t1,t2; tot=1; for(int i=1;i<n;i++) { scanf("%d%d",&t1,&t2); t1++;t2++; ru[t2]++;chu[t1]++; add(t1,t2,inf); } int TT=n+1,SS=0; for(int i=1;i<=n;i++) { add(SS,i,inf);add(i,TT,inf); } S=n+2;T=n+3; for(int i=1;i<=n;i++) { if(chu[i]>ru[i]) { add(i,T,chu[i]-ru[i]); } else if(ru[i]>chu[i]) { add(S,i,ru[i]-chu[i]); } } dinic(); add(TT,SS,inf); dinic(); printf("%d\n",f[tot]); return 0; }
bzoj 4464 : [Jsoi2013]旅行時的困惑