暑期集訓之暢通工程1

省政府“暢通工程”的目標是使全省任何兩個村莊間都可以實現公路交通(但不一定有直接的公路相連,只要能間接通過公路可達即可)。經過調查評估,得到的統計表中列出了有可能建設公路的若干條道路的成本。現請你編寫程序,計算出全省暢通需要的最低成本。 
Input測試輸入包含若干測試用例。每個測試用例的第1行給出評估的道路條數 N、村莊數目M ( < 100 );隨後的 N 
行對應村莊間道路的成本,每行給出一對正整數,分別是兩個村莊的編號,以及此兩村莊間道路的成本(也是正整數)。爲簡單起見,村莊從1到M編號。當N爲0時,全部輸入結束,相應的結果不要輸出。 
Output對每個測試用例,在1行裏輸出全省暢通需要的最低成本。若統計數據不足以保證暢通,則輸出“?”。 
Sample Input
3 3
1 2 1
1 3 2
2 3 4
1 3
2 3 2
0 100
Sample Output
3
?

基本上也是套模板做的,就模板內容來說沒啥大區別 經典的kru啥算法模板,但就是在?這個輸出上有點麻煩,但其實也就是根據看下在整個程序完成後有多少個根節點 如果根節點等於一個的話就代表可以連通 反之代表不能連通,輸出?(注意 一定是先判斷連通的情況 不然就WA了QAQ),代碼如下:

#include<stdio.h>
#include<string.h>
#include<algorithm>
#define N 100010
using namespace std;
int par[N];
int hight[N]; 
struct edge
{
    int u,v,cost;
};
edge G[N];
int E,V;
void Init_union_find(int n)
{
    for(int i=0; i<=n; i++)
    {
        par[i]=i;
        hight[i]=0;
    }
}
int find(int x)
{
    if(par[x]==x)
        return x;
    else
        return par[x]=find(par[x]);
}
void unite(int x,int y)
{
    x=find(x);
    y=find(y);
    if(x==y)
    return ;
    if(hight[x]<hight[y])
        par[x]=y;
    else
    {
        par[y]=x;
        if(hight[x]==hight[y])
        hight[x]++;
    }
}
bool same(int x,int y)
{
    return find(x)==find(y);
}
bool cmp(edge a,edge b)
{
    return a.cost<b.cost;
}
int kruskal()
{
    sort(G,G+E,cmp);
    Init_union_find(V);
    int ans=0;
    for(int i=0; i<E; i++)
    {
        edge e=G[i];
        if(!same(e.u,e.v))
        {
            unite(e.u,e.v);
            ans+=e.cost;
        }
    }
    return ans;
}
int main()
{
	while(scanf("%d %d",&E,&V)&&E)
	{   
		for(int i=0;i<E;i++)
		{
			scanf("%d %d %d",&G[i].u,&G[i].v,&G[i].cost);
		}
		int k;
		k=kruskal();
		int h=0;
		for(int i=1;i<=V;i++)
		{
			if(find(i)==i)
			h++;
		}
		if(h==1)
		printf("%d\n",k);
		else
		printf("?\n");
	}
	return 0;
}

總的來說 就是模板題


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