D - Constructing Roads POJ - 2421
题意:有一些村庄需要修一些道路是所有村庄都可以连接,不过有些道路已经修好了,问题最少还需要修建的道路长度是多少。
输入的第一行是一个N代表N个村庄,下面是一个N*N的矩阵,代表着i->j的距离,然后输出一个Q,接着有Q行,表示AB已经修建的村庄
思路:为了增加麻烦他们设定了一些已经修建的村庄,使用krusal做,把已经修建的边都连接上
#include<cstdio>
#include<algorithm>
using namespace std;
#define INF 3000
struct village
{
int a,b,cost;
}p[10000+5];
int parent[100+5];
bool com(village a,village b)
{
return a.cost<b.cost;
}
void init()
{
for(int i=1;i<=100+5;i++)
{
parent[i]=i;
}
}
int find(int x)
{
int r=x;
while(r!=parent[r]) //寻找根结点
r=parent[r];
int i=x,j;
while(i!=r) //压缩路径
{
j=parent[i];
parent[i]=r;
i=j;
}
return r;
}
int Union(int a,int b)
{
int x=find(a);
int y=find(b);
if(x!=y)
{
parent[x]=y;
return 1;
}
return 0;
}
int main()
{
int n,m,x,y;
while(~scanf("%d",&n))
{
int ans=0;
if(!n) break;
int k=1,c;
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
{
scanf("%d",&c);
p[k].a=i;
p[k].b=j;
p[k++].cost=c;
}
init();
scanf("%d",&m);
for(int i=1;i<=m;i++)
{
scanf("%d%d",&x,&y);
Union(x,y);
}
sort(p+1,p+k,com);
int cont=0;
for(int i=n+1;i<=k;i++)
{
if(Union(p[i].a,p[i].b))
{
ans+=p[i].cost;
cont++;
}
if(cont==n-1)
break;
}
printf("%d\n",ans);
}
return 0;
}
/*int road[100+5][100+5]; //邻接矩阵
int dis[100+5]; //最短距离判断
int vis[100+5]; //1代表加入u集合,0代表未加入
int num;
void prime()
{
for(int i=1;i<=ni++)
{
dis[i]=road[1][i];
vis[i]=0;
}
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%d%d",&a,&b);
}
int v;
for(int i=1;i<=num;i++)
{
int minn=INF;
for(int j=1;j<=num;j++)
{
if(vis[j]==0&&dis[j]<minn)
v=j;
}
vis[v]=1;
for(int j=1;j<=num;j++)
{
if(vis[j]==0&&dis[j]>road[v][j]])
dis[j]=road[v][j];
}
}
int ans=0;
for(int i=1;i<=num;i++)
{
ans+=dis[i];
}
printf("%d",ans);
}
int main()
{
int n;
while(~scanf("%d",&n))
{
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
road[i][j]=INF;
}
num=n;
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
scanf("%d",road[i][j]);
prime();
}
return 0;
}*/