codevs 1003 电话连线

Posted 2020-09-18

题目描述 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;
}