HDU1863（最小生成树）

题目大意：求MST。。。

两种做法均可。

#include<cstdio>
#include<algorithm>
#include<cmath>
#include<iostream>
#include<cstring>
#include<string>
using namespace std;
const int maxn=1111;
int fa[maxn];
int n,m;
int cnt;
struct node
{
    int u,v,w;
    bool operator<(const node &a)const
    {
        return w<a.w;
    }
}p[maxn];
void init()
{
    for(int i=0;i<m;i++)
    {
        fa[i]=i;
    }
}
int find_fa(int x)
{
    return (fa[x]==x)?fa[x]:fa[x]=find_fa(fa[x]);
}
int kruskal()
{
    int ans=0;
    cnt=0;
    for(int i=0;i<n;i++)
    {
        int fx=find_fa(p[i].u);
        int fy=find_fa(p[i].v);
        if(fx!=fy)
        {
            fa[fx]=fy;
            ans+=p[i].w;
            cnt++;
        }
        if(cnt==m-1)
            break;
    }
    return ans;
}
int main()
{
    while(scanf("%d%d",&n,&m)&&n)
    {
        init();
        for(int i=0;i<n;i++)
        {
            scanf("%d%d%d",&p[i].u,&p[i].v,&p[i].w);
        }
        sort(p,p+n);
        int ans=kruskal();
        if(cnt<m-1)
            printf("?\n");
        else
            printf("%d\n",ans);
    }
    return 0;
}
/*Kruskal*/

#include<cstdio>
#include<algorithm>
#include<cmath>
#include<iostream>
#include<cstring>
#include<string>
using namespace std;
const int maxn=222;
const int inf=0x3f3f3f3f;
int dis[maxn][maxn];
int d[maxn];
bool vis[maxn];
int n,m;
int ans;
void prim()
{
    ans=0;
    for(int i=1;i<=m;i++)
    {
        d[i]=dis[1][i];
    }
    d[1]=0;
    vis[1]=1;
    for(int i=1;i<m;i++)
    {
        int minx=inf;
        int num=-1;
        for(int j=1;j<=m;j++)
        {
            if(!vis[j]&&d[j]<minx)
            {
                minx=d[j];
                num=j;
            }
        }
        if(minx==inf)
        {
            printf("?\n");
            return;
        }
        vis[num]=1;
        ans+=minx;
        for(int j=1;j<=m;j++)
        {
            if(!vis[j]&&d[j]>dis[num][j])
            {
                d[j]=dis[num][j];
            }
        }
    }
    printf("%d\n",ans);
}
int main()
{
    while(scanf("%d%d",&n,&m)&&n)
    {
        int u,v,w;
        memset(vis,0,sizeof vis);
        memset(d,0,sizeof d);
        for(int i=1;i<=m;i++)
        {
            for(int j=1;j<=m;j++)
            {
                if(i==j)
                    dis[i][j]=0;
                else
                    dis[i][j]=inf;
            }
        }
        for(int i=0;i<n;i++)
        {
            scanf("%d%d%d",&u,&v,&w);
            dis[u][v]=dis[v][u]=w;
        }
        prim();
    }
    return 0;
}
/*Prim*/