Hdu 3371 Connect the Cities（最小生成树）

地址：http://acm.hdu.edu.cn/showproblem.php?pid=3371

 

其实就是最小生成树，但是这其中有值得注意的地方：就是重边。题目没有告诉你两个城市之间只有一条路可走，所以两个城市之间可能有多条路可以走。

举例： 输入可以包含 1 2 3  // 1到2的成本为3

　　　　　　　　　  1 2 5  //1到2的成本为5

　　　　　　　　     因此此时应选成本较低的路。

然后，已经连通的城市之间的连通成本为0。

 

这题用G++提交得到984ms的反馈，用C++提交则得到484ms的反馈。

很想知道这个时间差是怎么回事。

另外，这题一定还可以优化。

 

#include <iostream>

#include <cstring>

#include <cstdio>

using namespace std;



const int maxn = 500+10;

int dis[maxn][maxn]; //存两地间的距离

bool vis[maxn];      //是否已经加入树

int temp[maxn];

int shortP[maxn];

const int INF = 65523655;

int n,m,k;



inline void input() //初始及输入

{

    cin>>n>>m>>k;

    int i,j;

    for(i=1;i<=n;i++)

        for(j=1;j<=n;j++)

        dis[i][j]=INF;

    int p,q,r;

    while(m--)

    {

        scanf("%d%d%d",&p,&q,&r);

        if( r<dis[p][q] )        //这一句保证存入的是两地间距离的最小值

            dis[p][q]=dis[q][p]=r;

    }

    int t;

    memset(temp,0,sizeof(temp));

    while(k--)

    {

        scanf("%d",&t);

        for(i=1;i<=t;i++)

            scanf("%d",&temp[i]);

        for(i=1;i<=t;i++)

            for(j=i+1;j<=t;j++)

                dis[ temp[i] ][ temp[j] ]=dis[ temp[j] ][ temp[i] ]=0;

    }

    memset(vis,false,sizeof(vis));

    for(i=1;i<=n;i++)

        shortP[i]=INF;

}



inline int span() //求最低成本

{

    int cost=0;

    int pos=1,k;

    int count=0;

    int i,minC=INF;

    vis[1]=true;

    while(count<n-1) //判断最小生成树是否已经完成

    {

        for(i=1;i<=n;i++)

            if( !vis[i] && shortP[i] > dis[pos][i] )

                shortP[i]=dis[pos][i];

        k=-1;

        minC=INF;

        for(i=1;i<=n;i++)

            if( !vis[i] && minC > shortP[i] )

            {

                minC=shortP[i];

                k=i;

            }

        if(k==-1)

            return -1; //判断是否还可以再把一个城市加进来

        cost += shortP[k];

        vis[k]=true;

        pos=k;

        count++;

    }

    return cost;

}



int main()

{

    int T;

    while(cin>>T)

    {

        int flag;

        while(T--)

        {

            input();

            flag = span();

            cout<<flag<<endl;

        }

    }

    return 0;

}