题目描述 Description
一个国家有n个城市。若干个城市之间有电话线连接,现在要增加m条电话线(电话线当然是双向的了),使得任意两个城市之间都直接或间接经过其他城市有电话线连接,你的程序应该能够找出最小费用及其一种连接方案。
输入描述 Input Description
输入文件的第一行是n的值(n<=100).
第二行至第n+1行是一个n*n的矩阵,第i行第j列的数如果为0表示城市i与城市j有电话线连接,否则为这两个城市之间的连接费用(范围不超过10000)。
输出描述 Output Description
输出文件的第一行为你连接的电话线总数m,第二行至第m+1行为你连接的每条电话线,格式为i j,(i<j), i j是电话线连接的两个城市。输出请按照Prim算法发现每一条边的顺序输出,起始点为1.
第m+2行是连接这些电话线的总费用。
样例输入 Sample Input
5
0 15 27 6 0
15 0 33 19 11
27 33 0 0 17
6 19 0 0 9
0 11 17 9 0
样例输出 Sample Output
2
1 4
2 5
17
数据范围及提示 Data Size & Hint
n<=100
分析:
prim最小生成树的模版题。。。
刚开始写了个kruskle,才发现路线输出顺序不对QAQ
直接prim,距离不为0时,记录路线,最后输出即可。
#include<iostream> #include<algorithm> using namespace std; int a[110][110],dis[110],book[110]; int from[110]; int record[110][2]; const int inf=9999999; int main() { int n,m=1,ans=0,cnt=0; cin>>n; for(int i=1;i<=n;++i) for(int j=1;j<=n;++j)cin>>a[i][j]; for(int i=1;i<=n;++i)dis[i]=inf; dis[1]=0; for(int i=1;i<=n;++i) { int temp=inf,t; for(int j=1;j<=n;++j) { if(dis[j]<temp&&book[j]==0)temp=dis[j],t=j; } book[t]=1; ans+=temp; for(int j=1;j<=n;++j) { if(a[t][j]<dis[j]&&book[j]==0) { dis[j]=a[t][j]; from[j]=t; } } if(temp) { ++cnt; record[cnt][0]=min(t,from[t]); record[cnt][1]=max(t,from[t]); } } cout<<cnt<<endl; for(int i=1;i<=cnt;++i) { cout<<record[i][0]<<" "<<record[i][1]<<endl; } cout<<ans; return 0; }
时间: 2024-12-20 15:12:16