HDU 1233 (最小生成樹)

Description

某省調查鄉村交通狀況,得到的統計表中列出了任意兩村莊間的距離。省政府“暢通工程”的目標是使全省任何兩個村莊間都可以實現公路交通(但不一定有直接的公路相連,只要能間接通過公路可達即可),並要求鋪設的公路總長度爲最小。請計算最小的公路總長度。 

Input

測試輸入包含若干測試用例。每個測試用例的第1行給出村莊數目N ( < 100 );隨後的N(N-1)/2行對應村莊間的距離,每行給出一對正整數,分別是兩個村莊的編號,以及此兩村莊間的距離。爲簡單起見,村莊從1到N編號。 
當N爲0時,輸入結束,該用例不被處理。 

Output

對每個測試用例,在1行裏輸出最小的公路總長度。 

Sample Input

3
1 2 1
1 3 2
2 3 4
4
1 2 1
1 3 4
1 4 1
2 3 3
2 4 2
3 4 5
0

Sample Output

3
5


        
  
Huge input, scanf is recommended.

模板題。


1.普里姆算法:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;
#define sc(a) scanf("%d",&a)
#define nsc(a,b) scanf("%d %d",&a,&b)
#define rsc(a,b,c) scanf("%d %d %d",&a,&b,&c)
#define pr(a) printf("%d\n",a)
#define mem(a,x) memset(a,x,sizeof(a))
#define lf(i,l,r) for(int i=l;i<r;i++)
#define inf 0x3f3f3f3f

int n,m,sum,spot[105][105],vis[105],dis[105];

void solve()
{
    int maxn,tol;
    sum=0;
    lf(i,1,n+1)
    dis[i]=spot[1][i];
    vis[1]=1;
    lf(i,1,n)
    {
        tol=-1;
        maxn=inf;
        lf(j,1,n+1)
        {
            if(!vis[j]&&maxn>dis[j])
            {
                maxn=dis[j];
                tol=j;
            }
        }
        if(tol==-1)
            return ;
        sum+=maxn;
        vis[tol]=1;
        lf(i,1,n+1)
        if(!vis[i])
            dis[i]=min(dis[i],spot[tol][i]);
    }

}

int main()
{
    int a,b,v;
    while(~sc(n)&&n)
    {
        lf(i,1,n+1)
        {
            vis[i]=0;
            lf(j,1,n+1)
            {
                spot[i][j]=inf;
            }
        }
        lf(i,1,n*(n-1)/2+1)
        {
            rsc(a,b,v);
            spot[b][a]=spot[a][b]=v;
        }
        solve();
        pr(sum);
    }
    return 0;
}


2.克魯斯卡爾算法:

#include <iostream>  // 克魯斯卡爾算法
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;
#define sc(a) scanf("%d",&a)
#define nsc(a,b) scanf("%d %d",&a,&b)
#define rsc(a,b,c) scanf("%d %d %d",&a,&b,&c)
#define pr(a) printf("%d\n",a)
#define mem(a,x) memset(a,x,sizeof(a))
#define lf(i,l,r) for(int i=l;i<r;i++)
#define inf 0x3f3f3f3f

int n,m,sum,num,p[105];

struct node
{
    int s;
    int e;
    int v;
} spot[5005];

int cmp(node x,node y)
{
    return x.v<y.v;
}

int Find(int x)
{
    if(x!=p[x])
    {
        p[x]=Find(p[x]);
    }
    return p[x];
}

void solve()
{
    int tol=0;
    sum=0;
    lf(i,1,num)
    {
        int xx=Find(spot[i].s);
        int yy=Find(spot[i].e);
        if(xx!=yy)
        {
            p[xx]=yy;
            tol++;
            sum+=spot[i].v;
            if(tol==n-1)
                return ;
        }
    }
    return ;
}

int main()
{
    int a,b,v;
    while(~sc(n)&&n)
    {
        num=n*(n-1)/2+1;
        lf(i,1,n+1)
        p[i]=i;
        lf(i,1,num)
        {
            rsc(a,b,v);
            spot[i].s=a;
            spot[i].e=b;
            spot[i].v=v;
        }
        sort(spot+1,spot+num,cmp);
        solve();
        pr(sum);
    }
    return 0;
}


 



發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章