stoer wagner代碼

合併的代碼很容易寫錯!
#include <cstdio>
#include <string.h>
#include <cstdlib>
#include <algorithm>
using namespace std;
const int NMax=550;
int N,M,G[NMax][NMax],dist[NMax],map[NMax];
bool inA[NMax];
int Dinic() {
//1.min=MAXINT,固定一個頂點P 
//2.從點P用類似prim的s算法擴展出“最大生成樹”,記錄最後擴展的頂點和最後擴展的邊 
//3.計算最後擴展到的頂點的切割值(即與此頂點相連的所有邊權和),若比min小更新min 
//4.合併最後擴展的那條邊的兩個端點爲一個頂點(當然他們的邊也要合併,這個好理解吧?) 
//5.轉到2,合併N-1次後結束 
//6.min即爲所求,輸出min 
	int ret=(~0u>>1),s=1,pre;
	for(int i=1;i<=N;i++) map[i]=i;
	while(N>1) {
		memset(inA,0,sizeof(inA));
	    inA[map[s]]=1;
		int lastx,lasty;pre=s;
		for(int i=1;i<=N;i++) dist[map[i]]=G[map[s]][map[i]];
		for(int I=1;I<N;I++) {
			int next=-1;
			for(int i=2;i<=N;i++) if(!inA[map[i]]&&(next==-1||dist[map[i]]>dist[map[next]])) next=i;
			inA[map[next]]=1;
			for(int i=1;i<=N;i++) if(!inA[map[i]]) dist[map[i]]+=G[map[next]][map[i]];
			if(I==N-1) {lastx=pre;lasty=next;}
			pre=next;
		}
		ret=min(ret,dist[map[lasty]]);
		for(int i=1;i<=N;i++) G[map[lastx]][map[i]]=G[map[i]][map[lastx]]=G[map[lastx]][map[i]]+G[map[i]][map[lasty]];
//就是上一行,千萬不要寫成G[map[lastx]][map[lasty]]+G[map[i]][map[lasty]]
		map[lasty]=map[N--];
	}
	return ret==(~0u>>1)?0:ret;
}
int main()
{
	while(scanf("%d%d",&N,&M)!=EOF) {
		int x,y,z;
	    memset(G,0,sizeof(G));
		for(int i=1;i<=M;i++) {
			scanf("%d%d%d",&x,&y,&z);
			x++;y++;
			G[x][y]+=z;G[y][x]+=z;
		}
		printf("%d\n",Dinic());
	}
	return 0;
}


 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章