hdu_1233_畅通工程再续_最小生成树_kruscal_prim

本来差点都要放入”存疑“归档了，午休之后回来仔细检查一下代码，发现测试用例”1 1 2 6“通不过，结果是因为自己在遍历边时用的是

for(i=0;i<n;i++) 而边是有n*(n-1)/2个的。

每个人写代码一段时间后都会有固定的规范，比如这道题用并查集+kruscal就看到多种写法，有的人是多不喜欢main()里面的函数啊。

Kuruscal代码如下：

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=1000;
int pre[maxn];
struct edge
{
    int x;
    int y;
    int weight;
    bool operator<(const edge& cur)const
    {
        return weight<cur.weight;
    }
}edge[maxn*(maxn-1)/2];

int find(int x)
{
    int r=x;
    while(pre[r]!=r)
        r=pre[r];
    int i=x,j;
    while(pre[i]!=r)
    {
        j=pre[i];
        pre[i]=r;
        i=j;
    }
    return r;
}

bool judge(int x,int y)  //判断兼合并
{
    int fx=find(x);
    int fy=find(y);
    if(fx!=fy) {pre[fx]=fy;return true;}
    return false;
}
int main(void)
{
    int n,m;
    int a,b;
    int i,j,k;
    while(cin>>n&&n!=0)
    {
        for(i=0;i<n*(n-1)/2;i++)
            cin>>edge[i].x>>edge[i].y>>edge[i].weight;
        memset(pre,0,sizeof(pre));
        for(i=1;i<=n;i++) pre[i]=i;
        int weight_sum=0;
        sort(edge,edge+n*(n-1)/2);
        for(i=0;i<n*(n-1)/2;i++)   //这里出现的错误真是，，写代码的时候走心了
        {
            if(judge(edge[i].x,edge[i].y)) weight_sum+=edge[i].weight;
        }
        cout<<weight_sum<<endl;

    }
}

关于prim()算法，实际上就是选点，并且在每次选完点后更新<集合>到各个点的最短距离。

最开始时用dist[]来存储map[1][]（因为一定会选中一个点，那么不妨从第一个点开始)，之后每当选中一个新的点pos时，比较map[pos][]和dist[]的关系。

代码如下:

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=1000;
int map[maxn][maxn],visit[maxn],dist[maxn];
<pre name="code" class="cpp">const int INF=999999;

int n,m;

int prim()
{
    int i,j,k;
    int pos,sum=0;
    for(i=1;i<=n;i++) dist[i]=map[1][i];//初始化
    visit[1]=1;
    for(i=1;i<n;i++)
    {
        int min=INF;
        for(j=1;j<=n;j++)
        {
            if(!visit[j]&&dist[j]<min)
            {
                pos=j;
                min=dist[j];
            }
        }
        visit[pos]=1;
        for(j=1;j<=n;j++)
        {
            if(!visit[j]&&map[pos][j]<dist[j])
                dist[j]=map[pos][j]; //进行更新
        }
    }
    for(i=1;i<=n;i++) //<=?
        sum+=dist[i];
    return sum;
}
int main(void)
{
    int i,j,k;
    while(cin>>n)
    {
        if(n==0) break;
        memset(visit,0,sizeof(visit));

       /* for(i=1;i<n;i++)
            for(j=i+1;j<=n;j++)
        {
            cin>>map[i][j];
            map[j][i]=map[i][j];
        }*/
        int x,y,z;
        for(i=1;i<=n*(n-1)/2;i++)
        {
            cin>>x>>y>>z;
            map[x][y]=map[y][x]=z;
        }
        for(i=1;i<=n;i++) map[i][i]=0; //说明无法访问
        cout<<prim()<<endl;
    }
}