nyoj 118 修路方案

描述

南将军率领着许多部队，它们分别驻扎在N个不同的城市里，这些城市分别编号1~N，由于交通不太便利，南将军准备修路。

现在已经知道哪些城市之间可以修路，如果修路，花费是多少。

现在，军师小工已经找到了一种修路的方案，能够使各个城市都联通起来，而且花费最少。

但是，南将军说，这个修路方案所拼成的图案很不吉利，想让小工计算一下是否存在另外一种方案花费和刚才的方案一样，现在你来帮小工写一个程序算一下吧。

输入

第一行输入一个整数T(1<T<20)，表示测试数据的组数
每组测试数据的第一行是两个整数V,E，(3<V<500,10<E<200000)分别表示城市的个数和城市之间路的条数。数据保证所有的城市都有路相连。
随后的E行，每行有三个数字A B L，表示A号城市与B号城市之间修路花费为L。

输出

对于每组测试数据输出Yes或No（如果存在两种以上的最小花费方案则输出Yes,如果最小花费的方案只有一种，则输出No)

样例输入

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

样例输出

No
Yes

思路：这道题让找两种以上的最小花费方案，需要找次小生成树，（1）先找出最小生成树，并把组成最小生成树的边标记，（2）然后每次删除最小生成树中的一条边，看其他边组成的最小生成树等不等与第一次算的，如果等于就可以break，说明已经找了两种最小方案（3）但是要注意可能删除一个边后其他边组不成一个连通的通路

#include<stdio.h>
#include<algorithm>
#include<string.h>
using namespace std;

int f[510],p[200010];
int n,m;
struct node
{
    int u;
    int v;
    int w; //花费
} e[200010];

bool cmp(node n1,node n2)
{
    return n1.w<n2.w;
}

int getf(int v)
{
    if(f[v]==v)
        return v;
    f[v]=getf(f[v]);
    return f[v];
}
int merge(int u,int v)
{
    int t1,t2;
    t1=getf(f[u]);
    t2=getf(f[v]);
    if(t1!=t2)
    {
        f[t2]=t1;
        return 1;
    }
    return 0;
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&m);
        int sum=0,num=0,flag;
        for(int i=1; i<=n; i++)
        {
            f[i]=i;
        }
        memset(p,0,sizeof(p));
        for(int i=1; i<=m; i++)
            scanf("%d%d%d",&e[i].u,&e[i].v,&e[i].w);
        sort(e+1,e+m+1,cmp);
        for(int i=1; i<=m; i++)
        {
            if(merge(e[i].u,e[i].v))
            {
                sum+=e[i].w;
                num++;
                p[i]=1;
            }
            if(num==n-1)
                break;
        }
        for(int j=1; j<m; j++)
        {
            flag=0;
            int sum1=0,num1=0;
            if(p[j])
            {
                for(int i=1; i<=n; i++)
                    f[i]=i;
                for(int i=1; i<=m; i++)
                {
                    if(i!=j)
                    {
                        if(merge(e[i].u,e[i].v))
                        {
                            sum1+=e[i].w;
                            num1++;
                        }
                        if(num1==n-1)
                            break;
                    }
                }
            }
            if(sum1==sum&&num1==n-1)
            {
                flag=1;
                printf("Yes\n");
                break;
            }

        }
        if(flag==0)
            printf("No\n");
    }
    return 0;
}