HDU 5631 Rikka with Graph (并查集)

题目:http://acm.hdu.edu.cn/showproblem.php?pid=5631

代码:

#include<stdio.h>
#include<string.h>
using namespace std;
int u[121];
int v[121];
int f[121];
int n;
int output;
int find(int x)
{
    return f[x] == x ? x : (f[x] = find(f[x]));
}
void merge(int a,int b)
{
    int A,B;
    A=find(a);
    B=find(b);
    if(A!=B)
    f[B]=A;
}
void init()
{
    for(int i=0;i<=n;i++)
    {
        f[i]=i;
    }
}
void solve(int x)
{
    init();
    for(int i=0;i<n+1;i++)
    {
        if(i!=x)
        {
            merge(u[i],v[i]);
        }
    } 
    int cont=0;
    for(int i=1;i<=n;i++)
    {
        if(f[i]==i)
            cont++;
    }
    if(cont==1)
        output++;
}
void solve2(int x,int y)
{
    init();
    for(int i=0;i<n+1;i++)
    {
        if(i!=x&&i!=y)
        {
            merge(u[i],v[i]);
        }
    }
    int cont=0;
    for(int i=1;i<=n;i++)
    {
        if(f[i]==i)
        cont++;
    }
    if(cont==1)
    output++;
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        for(int i=0;i<n+1;i++)
        {
            scanf("%d%d",&u[i],&v[i]);
        }
        output=0;
        for(int i=0;i<n+1;i++)
        {
            solve(i);
        }
        for(int i=0;i<n+1;i++)
        {
            for(int j=i+1;j<n+1;j++)
            {
                solve2(i,j);
            }
        }
        printf("%d\n",output);
    }
}



大神的代码,直接抄,再分析。

让 nn 个点联通最少需要 n-1n1 条边,所以最多只能删除两条边,我们可以枚举删除的这两条边(或者唯一的一条边),然后暴力BFS判断连通性就好了。时间复杂度 O(n^3)O(n3)

你可能感兴趣的:(HDU 5631 Rikka with Graph (并查集))