最小割集Stoer-Wagner算法，网络最大流问题

1#include <iostream>
2#include <cmath>
3using namespace std;
4int mat[600][600];
5int res;
6//Stoer-Wagner算法，加了自己看得懂的备注
7//无向图全局最小割，用求prim类似方法o（n^3)，学习了一个下午……
8//一开始用枚举源点汇点的最大流求解，复杂度o(n^5) 超时
9int min(int a,int b)
10{
11    if(a<b) return a;
12    else return b;
13}
14
15void Mincut(int n)
16{
17    int node[600], dist[600];
18    bool visit[600];
19    int i, prev, j, k;
20    for (i = 0; i < n; i++)
21        node[i] = i;
22    while (n > 1)
23    {
24        int maxj = 1;
25        for (i = 1; i <n; i++) //初始化到已圈集合的割大小
26    {
27            dist[node[i]] = mat[node[0]][node[i]];
28            if (dist[node[i]] > dist[node[maxj]])
29                maxj = i;
30        }
31        prev = 0;
32        memset(visit, false, sizeof (visit));
33        visit[node[0]] = true;
34        for (i = 1; i < n; i++)
35    {
36                        if (i == n - 1)//只剩最后一个没加入集合的点，更新最小割
37            {
38                                res = min(res, dist[node[maxj]]);
39                                for (k = 0; k < n; k++) //合并最后一个点以及推出它的集合中的点
40                {
41                    mat[node[k]][node[prev]]=(mat[node[prev]][node[k]]+=mat[node[k]][node[maxj]]);
42                }
43                                node[maxj] = node[--n]; //缩点后的图
44                        }
45                        visit[node[maxj]] = true;
46                        prev = maxj;
47                        maxj = -1;
48                        for (j = 1; j < n; j++)
49            {
50                                if (!visit[node[j]]) //将上次求的maxj加入集合，合并与它相邻的边到割集
51                {
52                                      dist[node[j]] += mat[node[prev]][node[j]];
53                                      if (maxj == -1 || dist[node[maxj]] < dist[node[j]])
54                                      maxj = j;
55                                }
56            }
57        }
58
59    }
60    return;
61}
62
63int main()
64 {
65    int n, m, a, b, v;
66    while (scanf("%d%d", &n, &m) != EOF)
67    {
68        res = (1 << 29);
69        memset(mat, 0, sizeof (mat));
70        while (m--)
71    {
72            scanf("%d%d%d", &a, &b, &v);
73            mat[a][b] += v;
74            mat[b][a] += v;
75        }
76        Mincut(n);
77        printf("%d\n", res);
78    }
79    return 0;
80}
81