题目:Flow Problem
#include <iostream> #include <stdio.h> #include <string.h> using namespace std; const int N=2050; //N表示点的数量 const int M=2500000; //M表示边的数量,注意应大于(N-1)*N,因为网络是双向的 const int INF=0x7fffffff; int e; int ver[M],next[M],flow[M]; //ver为边指向的节点,flow为边的容量,next为链表的下一条边 int head[N],work[N],dis[N],q[N]; //head为节点的链表头,work 用于算法中的临时链表头,dis 计算距离 void init() { e=0; memset(head,-1,sizeof(head)); } void add(int u,int v,int c) //增加一条从u到v容量为c的边 { ver[e]=v;flow[e]=c;next[e]=head[u];head[u]=e++; ver[e]=u;flow[e]=0;next[e]=head[v];head[v]=e++; } //广搜计算出每个点与源点的最短距离,如果不能到达汇点说明算法结束 int bfs(int S,int T) { int rear=0; memset(dis,-1,sizeof(dis)); dis[S]=0;q[rear++]=S; for(int i=0;i<rear;i++) { for(int j=head[q[i]];j!=-1;j=next[j]) { if(flow[j]&&dis[ver[j]]==-1) { dis[ver[j]]=dis[q[i]]+1; q[rear++]=ver[j]; if(ver[j]==T) return 1; } } } return 0; } //寻找可行流的增广路算法,按节点的距离来找,加快速度 int dfs(int cur,int a,int T) { if(cur==T) return a; for(int &i=work[cur];i!=-1;i=next[i]) //work是临时链表头,这里用i引用它,这样寻找过的边不再寻找 { if(flow[i]&&dis[ver[i]]==dis[cur]+1) { int t=dfs(ver[i],min(a,flow[i]),T); if(t) { flow[i]-=t; flow[i^1]+=t; //正反向边容量改变 return t; } } } return 0; } //求最大流,直到没有可行流 long long Dinic(int S,int T) { long long ans=0; while(bfs(S,T)) { memcpy(work,head,sizeof(head)); while(int t=dfs(S,INF,T)) ans+=t; } return ans; } int main() { long long sum; int t,n,m,S,T,x,y,c,k=1; scanf("%d",&t); while(t--) { scanf("%d%d",&n,&m); init(); S=1;T=n; while(m--) { scanf("%d%d%d",&x,&y,&c); add(x,y,c); } printf("Case %d: %I64d\n",k++,Dinic(S,T)); } return 0; }