XTU 1045 唯一的最小生成树

唯一的最小生成树
[ Submit Code ] [ Top 20 Runs ]
Acceteped : 227		Submit : 1065
Time Limit : 1000 MS		Memory Limit : 65536 KB
Description
求一个非负权边的无向连通图的最小生成树，如果这个无向图存在两个或两个以上的最小生成树，就输出Not Unique，否则输出最小生成树的边的权值和。 输入： 第一行是一个整数K，表示有多少个测试用例，以后每个测试用例占m+1行。每个测试用例的第一行为两个整数n，m（3<=n<=100），表示图的顶点数和边数，从第二行开始每行为三个整数i，j,w,表示从i到j顶点的权值。 输出： 每行输出一个测试用例的结果。如果这个无向图存在两个或两个以上的最小生成树，就输出Not Unique，否则输出最小生成树的边的权值和。
Sample Input
2 3 3 1 2 1 2 3 2 3 1 3 4 4 1 2 2 2 3 2 3 4 2 4 1 2
Sample Output
3 Not Unique
Source
程序设计实践
[ Submit Code ] [ Top 20 Runs ]

思路：要判断是否存在另一棵最小生成树，那么只要找出其中的一棵，然后逐条边地屏蔽，判断是否存在另一棵，如果存在则不是唯一的，否则就是唯一的。

AC代码如下：

#include <bits/stdc++.h>
//#pragma comment(linker, "/STACK:102400000, 102400000")
using namespace std;
const int maxn = 105;
const int maxm = 1005;
struct edge{
    int u, v, w;
    bool operator < (const edge & p) const {
        return w < p.w;
    }
} edges[maxm];
int p[maxn];
bool vis[maxm];
int n, m;

void init(){
    for (int i=1; i<=n; ++i)
        p[i] = i;
}

int find(int x){
    return p[x]==x ? x : p[x]=find(p[x]);
}

int kruskal(){
    int ans=0, cnt=0;
    init();
    for (int i=0; i<m; ++i){
        int x = find(edges[i].u);
        int y = find(edges[i].v);
        if (x != y){
            ans += edges[i].w;
            p[x] = y;
            ++cnt;
            vis[i] = true;
            if (cnt == n-1)
                break;
        }
    }
    return ans;
}

int unique(int x){
    int ans=0, cnt=0;
    init();
    for (int i=0; i<m; ++i){
        if (i == x)
            continue;
        int x = find(edges[i].u);
        int y = find(edges[i].v);
        if (x != y){
            ans += edges[i].w;
            p[x] = y;
            ++cnt;
            if (cnt == n-1)
                break;
        }
    }
    return ans;
}

int main(){
    #ifdef LOCAL
    // freopen("input.txt", "r", stdin);
    // freopen("output.txt", "w", stdout);
    #endif
    int T;
    scanf("%d", &T);
    while (T--){
        scanf("%d%d", &n, &m);
        memset(vis, false, sizeof(vis));
        for (int i=0; i<m; ++i)
            scanf("%d%d%d", &edges[i].u, &edges[i].v, &edges[i].w);
        sort(edges, edges+m);
        int ans = kruskal();
        bool ok = true;
        for (int i=0; i<m; ++i)
            if (vis[i]){
                int res = unique(i);
                if (ans == res){
                    ok = false;
                    break;
                }
            }
        if (ok)
            printf("%d\n", ans);
        else
            printf("Not Unique\n");
    }
    return 0;
}

添加一份更简洁的代码。

#include <bits/stdc++.h>
using namespace std;
const int maxn = 105;
const int maxm = 1005;
struct edge{
    int u, v, w;
    bool used, del;
    bool operator < (const edge & p) const {
        return w < p.w;
    }
} edges[maxm];
int p[maxn];
bool first;
int n, m;

void init(){
    for (int i=1; i<=n; ++i)
        p[i] = i;
}

int _find(int x){
    return p[x]==x ? x : p[x]=_find(p[x]);
}

int kruskal(){
    int ans = 0;
    init();
    for (int i=0; i<m; ++i){
        if (edges[i].del)
            continue;
        int x=_find(edges[i].u), y=_find(edges[i].v);
        if (x != y){
            ans += edges[i].w;
            p[x] = y;
            if (first)
                edges[i].used = true;
        }
    }
    return ans;
}

int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    int T;
    scanf("%d", &T);
    while (T--){
        scanf("%d%d", &n, &m);
        for (int i=0; i<m; ++i){
            scanf("%d%d%d", &edges[i].u, &edges[i].v, &edges[i].w);
            edges[i].del = false;
            edges[i].used = false;
        }
        sort(edges, edges+m);
        first = true;
        int ans = kruskal();
        bool ok = true;
        first = false;
        for (int i=0; i<m; ++i)
            if (edges[i].used){
                edges[i].del = true;
                int res = kruskal();
                if (ans == res){
                    ok = false;
                    break;
                }
                edges[i].del = false;
            }
        if (ok)
            printf("%d\n", ans);
        else
            printf("Not Unique\n");
    }
    return 0;
}