URAL-1416 (典型的次小生成树)

来源：互联网发布：matlab中初始化矩阵编辑：程序博客网时间：2024/06/08 06:16

题目大意：

给出一个带权无向图，求出最小生成树和次小生成树，注意这里两个结果可以相等。当生成树不存在的时候就可以输出-1；

思路：

这是典型的次小生成树的求法，注意细节尤其是初始化。错了好多次。
次小生成树就是有一条边和最小生成树不一样。所以保存最小生成树的每一条边，然后遍历即可。

#include <iostream>#include <cstdio>#include <cstring>#include <algorithm>using namespace std;const int MAX = 505;const int INF = 0x3f3f3f3f;int used[MAX];int f[MAX];int n,m;int Minused;struct Edge{    int s,e,v;}edge[MAX*MAX];int cmp(Edge a,Edge b){    return a.v < b.v;}void init(){    for(int i = 0;i <= n; i++)        f[i] = i;}int Find(int x){    if(x != f[x])        f[x] = Find(f[x]);    return f[x];}int krustal(){    init();    memset(used,0,sizeof(used));    sort(edge,edge+m,cmp);    int pos = 0,flag = 0;    Minused = 0;    int ans = 0;    for(int i = 0;i < m; i++) {        int a = Find(edge[i].s);        int b = Find(edge[i].e);        if(a != b) {            f[a] = b;            pos++;            used[Minused++] = i;            ans += edge[i].v;        }        if(pos == n - 1) {            break;        }    }    if(pos == n - 1)        return ans;    return -1;}int Next_min_tree(){    int temp = INF;    for(int i = 0;i < Minused; i++) {        init();        int pos = 0;        int ans = 0;        for(int j = 0;j < m; j++) {            if(used[i] != j) {                int a = Find(edge[j].s);                int b = Find(edge[j].e);                if(a != b) {                    f[a] = b;                    pos++;                    ans += edge[j].v;                }                if(pos == n -1)                    break;            }        }        if(pos == n - 1 && temp > ans)            temp = ans;    }    if(temp == INF)        return -1;    return temp;}int main(){    //freopen("in.txt","r",stdin);    while(scanf("%d%d",&n,&m) != EOF) {        for(int i = 0;i < m; i++) {            scanf("%d%d%d",&edge[i].s,&edge[i].e,&edge[i].v);        }        int ans = krustal();        int ans1 = Next_min_tree();        printf("Cost: %d\n",ans);        printf("Cost: %d\n",ans1);    }    return 0;}

0 0