码迷,mamicode.com
首页 > 其他好文 > 详细

Constructing Roads POJ - 2421 最小生成树板子题

时间:2020-01-29 14:22:30      阅读:62      评论:0      收藏:0      [点我收藏+]

标签:clu   poj   cti   algo   bsp   out   sizeof   int   stream   

#include<iostream>
#include<cstring> 
#include<algorithm>
using namespace std;
const int N=110;
int p[N];
struct edge{
    int a;
    int b;
    int w;
}e[N*N];
int map[N][N],flag[N][N],num,n;
bool cmp(edge a,edge b)
{
    return a.w<b.w;
}
int find(int x)
{
    if(p[x]!=x)
        p[x]=find(p[x]);
    return p[x];
}
int main()
{
    while(cin>>n&&n)
    {
        memset(flag,0,sizeof flag);
        memset(map,0,sizeof map);
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                cin>>map[i][j];
        int m,x,y;
        cin>>m;
        while(m--)
        {
            cin>>x>>y;
            flag[x][y]=1;
        }
        num=0;
        for(int i=1;i<=n;i++)
            for(int j=i+1;j<=n;j++)
                if(flag[i][j])
                {
                    e[num].a=i;
                    e[num].b=j;
                    e[num++].w=0;
                }
                else
                {
                    e[num].a=i;
                    e[num].b=j;
                    e[num++].w=map[i][j];
                }
        sort(e,e+num,cmp);
        for(int i=1;i<=n;i++)
            p[i]=i;
        int sum=0;
        for(int i=0;i<num;i++)
        {
            int a=find(e[i].a);
            int b=find(e[i].b);
            int w=e[i].w;
            if(a!=b)
            {
                p[a]=b;
                sum+=w;
            }
        }
        cout<<sum<<endl;
    }
    return 0;
}

 

Constructing Roads POJ - 2421 最小生成树板子题

标签:clu   poj   cti   algo   bsp   out   sizeof   int   stream   

原文地址:https://www.cnblogs.com/QingyuYYYYY/p/12240099.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!