【杭电1233 -- 还是畅通工程】 (最小生成树)

xiaoxiao2021-02-28  97

 

某省调查乡村交通状况,得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可),并要求铺设的公路总长度为最小。请计算最小的公路总长度。   Input 测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( < 100 );随后的N(N-1)/2行对应村庄间的距离,每行给出一对正整数,分别是两个村庄的编号,以及此两村庄间的距离。为简单起见,村庄从1到N编号。 当N为0时,输入结束,该用例不被处理。   Output 对每个测试用例,在1行里输出最小的公路总长度。   Sample Input 3 1 2 1 1 3 2 2 3 4 4 1 2 1 1 3 4 1 4 1 2 3 3 2 4 2 3 4 5 0   Sample Output 3 5 Hint Hint Huge input, scanf is recommended.     

刚开始没注意题目数组开小了,WA了N次却不知道什么原因,好气呀,以后得吸取教训,反正数组开大点又不要钱。。

    

 #include<cstdio>

#include<cstring>

#include<algorithm> using namespace std; int p[10005]; struct road { int st; int end; int leng; }md[10005]; bool cmp(road a,road b) { return a.leng<b.leng; } int find(int x) { if(x!=p[x]) p[x]=find(p[x]); return p[x]; } void join(int x,int y) { int fx=find(x),fy=find(y); if(fx!=fy) { if(fy<fx) p[fx]=fy; else p[fy]=fx; } } int main() { int sum,n,i,ans; while(scanf("%d",&n)&&n) { int m=n*(n-1)/2; for(i=1;i<=n;i++) p[i]=i; for(i=1;i<=m;i++)    { scanf("%d%d%d",&md[i].st,&md[i].end,&md[i].leng); } sum=0; sort(md+1,md+m+1,cmp); for(i=1;i<=m;i++) { if(find(md[i].st)!=find(md[i].end))      //防止出现环  { join(md[i].st,md[i].end); sum+=md[i].leng;   } } printf("%d\n",sum); } return 0; }

转载请注明原文地址: https://www.6miu.com/read-28714.html

最新回复(0)