HDU 2122

题目大意很简单

就是给你城市的数量，和可以修建的铁路及其长度，如果连通，输出最小的总长，否则输出impossible

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=2122

我用的prim算法一直报错也不知道为什么，后来改用Kruscal算法就好了~~T T；

正确代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include<algorithm>
using namespace std;
#define N 1010
#define LL long long

int n,M,S,T,C,t,k;
LL ans;
int fa[N];

struct Path{
    int x,y,d;
    bool operator<(const Path &m)const{
        return d<m.d;
    }
}path[10010];

int getHead(int x)
{
    int a=x;
    while(x!=fa[x]) x=fa[x];
    fa[a]=x;
    return x;
}

bool Union(int x,int y)
{
    int fa_x=getHead(x);
    int fa_y=getHead(y);
    if(fa_x==fa_y) return false;
    fa[fa_x]=fa_y;
    return true;
}

int main()
{
    while(scanf("%d%d",&n,&M)!=EOF){
        t=0,k=0,ans=0;
        for(int i=0;i<n;i++) fa[i]=i;
        if(n==0) {cout<<0<<endl;continue;}
            //cout<<MAXN<<endl;
        for(int i=0;i<M;i++)  scanf("%d%d%d",&S,&T,&C),path[k].x=S,path[k].y=T,path[k++].d=C;
        sort(path,path+k);

        for(int i=0;i<k;i++){
            if(Union(path[i].x,path[i].y))
                ans+=path[i].d,t++;
        }

        if(t==n-1) cout<<ans<<endl;
        else cout<<"impossible"<<endl;
        cout<<endl;
    }
    return 0;
}

错误的还未能改正好的代码:

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
#define N 1010
#define MAXN 1000000000
#define LL long long
int n,M,S,T,C,count;
LL ans;
int visit[N],d[N];
int map[N][N];

void prim()
{
    int Min,MinIndex;
    memset(visit,0,sizeof(visit));
    ans=0;
    visit[0]=1,count=1;
    for(int i=1;i<n;i++) d[i]=map[0][i];

    for(int i=0;i<n-1;i++)
    {
        Min=MAXN,MinIndex=1;

        for(int i=1;i<n;i++)
            if(!visit[i]&&Min>d[i])
                Min=d[i],MinIndex=i;

        visit[MinIndex]=1;
        if(Min<MAXN) ans+=Min,count++;//cout<<Min<<' ';
        //cout<<ans<<endl;
        for(int i=1;i<n;i++)
            if(!visit[i]&&d[i]>map[i][MinIndex])
                d[i]=map[i][MinIndex];
    }
}

int main()
{
    while(scanf("%d%d",&n,&M)!=EOF){
        if(n==0) {cout<<0<<endl;continue;}
            //cout<<MAXN<<endl;
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++)
                map[i][j]=MAXN;
        }
        for(int i=0;i<M;i++)  scanf("%d%d%d",&S,&T,&C),map[S][T]=C,map[T][S]=C;;

        prim();

        /*for(int i=0;i<n;i++){
            for(int j=0;j<n;j++)
                cout<<map[i][j]<<' ';
            cout<<endl;
        }*/

        if(count==n) cout<<ans<<endl;
        else cout<<"impossible"<<endl;
        cout<<endl;
    }
    return 0;
}