POJ 2914 无向图最小割

以前总结过无向图最小割的最大流算法，n^2的枚举起点和终点，这个题必然tle

于是又学习了专门的算法：Stoer-Wagner算法

 

就是利用prim求最大生成树，将最后扫到的两点合并，执行n-1次，最终合并成一个点

参考资料：

http://kanboxshare.com/link/gCrQIYPJvdwCxGlOUwU5U3yxNNIiW2cUxIhddx7T60uYmiiO2heSC1rULa5tjEbf3MLRsM8QRKYGzoZhd8abNaUke

 

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <algorithm>
#include <iostream>

#define N 600
#define INF 1e9

using namespace std;

int map[N][N],q[N],dis[N];
bool vis[N],cut[N];
int n,m,S,T;

inline void read()
{
    memset(map,0,sizeof map);
    memset(cut,0,sizeof cut);
    for(int i=1,a,b,c;i<=m;i++)
    {
        scanf("%d%d%d",&a,&b,&c);
        map[a][b]+=c; map[b][a]+=c;
    }
}
//Stoer-Wagner算法
//prim求出最大生成树，将最后扫到的两点合并，执行n-1次 
inline int mincut(int now) 
{
    int tot=0;
    memset(vis,0,sizeof vis);
    memset(dis,0,sizeof dis);
    for(int j=0,tmp,k;j<now;j++)
    {
        tmp=-INF;
        for(int i=0;i<n;i++)
            if(!cut[i]&&!vis[i]&&tmp<dis[i]) tmp=dis[i],k=i;
        q[++tot]=k; vis[k]=true;
        for(int i=0;i<n;i++)
            if(!cut[i]&&!vis[i]) dis[i]+=map[k][i];
    }
    S=q[tot-1]; T=q[tot];
    return dis[T];
}

inline void contract()
{
    map[S][T]=map[T][S]=0;
    cut[T]=true;
    for(int i=0;i<n;i++)
        if(i!=S&&!cut[i]) map[i][S]=map[S][i]=map[S][i]+map[T][i];
}

inline void go()
{
    int ans=INF;
    for(int i=n;i>1;i--)
    {
        ans=min(ans,mincut(i));
        contract();
    }
    printf("%d\n",ans);
}

int main()
{
    while(scanf("%d%d",&n,&m)!=EOF) read(),go();
    return 0;
}