POJ 2421 Constructing Roads

//思路:把联通了的两个点的权值标记成0,然后用克鲁斯卡尔算法过或者其他的算法也行//

AC代码:

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
#define N 200005
int n,p[N],L,a[200][200];
struct node
{
    int w,v,c;
}t[N];
int cmp(node p,node q)
{
    return p.c<q.c;
}
int find(int k)
{
    if(p[k]==k)
    {
        return p[k];
    }
    else
    {
        return find(p[k]);
    }
}
int cl()
{
    int sum=0,cnt=0;
    int i;
    for(i=0;i<L;i++)
    {
        int x=find(t[i].w);
        int y=find(t[i].v);
        if(x!=y)
        {
            p[y]=x;
            cnt++;
            sum+=t[i].c;
        }
        if(cnt==n-1)
            return  sum;
    }
    return sum;
}
int main()
{
    scanf("%d",&n);
    int i,j,r;
    L=0;
    for(i=1;i<=n;i++)
    {
        for(j=1;j<=n;j++)
        {
            scanf("%d",&a[i][j]);
        }
        p[i]=i;
    }
    int m;
    scanf("%d",&m);
    int x,y;
    for(i=1;i<=m;i++)
    {
        scanf("%d%d",&x,&y);
        a[x][y]=0;
        a[x][y]=0;
    }
    for(i=1;i<=n;i++)
    {
        for(j=1;j<=n;j++)
        {
            t[L].w=i;
            t[L].v=j;
            t[L].c=a[i][j];
            L++;
        }
    }
    sort(t,t+L,cmp);
    printf("%d\n",cl());
}

你可能感兴趣的:(POJ 2421 Constructing Roads)