畅通工程 行电1232最小生成树

Problem Description

某省调查城镇交通状况，得到现有城镇道路统计表，表中列出了每条道路直接连通的城镇。省政府“畅通工程”的目标是使全省任何两个城镇间都可以实现交通（但不一定有直接的道路相连，只要互相间接通过道路可达即可）。问最少还需要建设多少条道路？ 

 

Input

测试输入包含若干测试用例。每个测试用例的第1行给出两个正整数，分别是城镇数目N ( < 1000 )和道路数目M；随后的M行对应M条道路，每行给出一对正整数，分别是该条道路直接连通的两个城镇的编号。为简单起见，城镇从1到N编号。 
注意:两个城市之间可以有多条道路相通,也就是说
3 3
1 2
1 2
2 1
这种输入也是合法的
当N为0时，输入结束，该用例不被处理。 

 

Output

对每个测试用例，在1行里输出最少还需要建设的道路数目。 

 

Sample Input

4 21 34 33 31 21 32 35 21 23 5999 00

 

Sample Output

102998
Hint
Hint
 Huge input, scanf is recommended.

 

Source

浙大计算机研究生复试上机考试-2005年

 

我的思路，这是最小生成树的小的变形，我的思路，把已经有路的路径长度设置为0，其他没路的权值设置为1，这样求解最小生成树，可以直接套用模板，而且最好的最小生成树的大小就是所求的所需的路的个数。

#include<iostream>
#include<algorithm>
#include<cstdio>
using namespace std;
struct line{
int st,ed,len;
};
line lines[1000001];
int p[1000001];
int map[1001][1001];
bool cmp(line a,line b)
{
if(a.len<b.len)
return true;
return false;
}
int find(int x)
{
return x==p[x]?x:p[x]=find(p[x]);
}
int main()
{
int n,m;
while(cin>>n)
{
int ans=0;
int t=0;
int count;
if(!n)break;
cin>>m;
for(int i=0;i<=n;i++)
for(int j=0;j<=n;j++)
map[i][j]=1;
for(int i=1;i<1001;i++)
p[i]=i;
for(int i=1;i<=m;i++)
{
int x,y;
scanf("%d%d",&x,&y);
map[x][y]=0;
map[y][x]=0;
}
for(int i=1;i<=n;i++)
for(int j=i+1;j<=n;j++)
{
count=t++;
lines[count].st=i;
lines[count].ed=j;
lines[count].len=map[i][j];
}
sort(lines,lines+t,cmp);
for(int i=0;i<t;i++)
{
int _x=find(lines[i].st);
int _y=find(lines[i].ed);
if(_x!=_y)
{
ans+=lines[i].len;
p[_x]=_y;
}
}
cout<<ans<<endl;
}
return 0;
}