1. 程式人生 > >ccf 最優灌溉

ccf 最優灌溉

<div class="pdsec">問題描述</div><div class="pdcont">  雷雷承包了很多片麥田,為了灌溉這些麥田,雷雷在第一個麥田挖了一口很深的水井,所有的麥田都從這口井來引水灌溉。
  為了灌溉,雷雷需要建立一些水渠,以連線水井和麥田,雷雷也可以利用部分麥田作為“中轉站”,利用水渠連線不同的麥田,這樣只要一片麥田能被灌溉,則與其連線的麥田也能被灌溉。
  現在雷雷知道哪些麥田之間可以建設水渠和建設每個水渠所需要的費用(注意不是所有麥田之間都可以建立水渠)。請問灌溉所有麥田最少需要多少費用來修建水渠。</div><div class="pdsec">輸入格式</div><div class="pdcont">  輸入的第一行包含兩個正整數n, m,分別表示麥田的片數和雷雷可以建立的水渠的數量。麥田使用1, 2, 3, ……依次標號。
  接下來m行,每行包含三個整數a<sub>i</sub>, b<sub>i</sub>, c<sub>i</sub>,表示第a<sub>i</sub>片麥田與第b<sub>i</sub>片麥田之間可以建立一條水渠,所需要的費用為c<sub>i</sub>。</div><div class="pdsec">輸出格式</div><div class="pdcont">  輸出一行,包含一個整數,表示灌溉所有麥田所需要的最小費用。</div><div class="pdsec">樣例輸入</div><div class="pddata">4 4
1 2 1
2 3 4
2 4 2
3 4 3</div><div class="pdsec">樣例輸出</div><div class="pddata">6</div>//最小生成樹
#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
using namespace std;
int G[1001][1001];
int n,m,vis[1001],dis[1001];
int p(){
	int cost=0;
	vis[1]=1;dis[1]=0;
	for(int i=1;i<=n;i++){
		if(G[1][i]){
			dis[i]=G[1][i];
		}
	}
	for(int k=0;k<n-1;k++){
		int s,min=0x3f3f3f3f;
		for(int i=1;i<=n;i++){
			if(min>dis[i]&&(!vis[i])){
				min=dis[i];
				s=i;
			}
		}
		cost+=min;
		vis[s]=1;
	//	cout<<s<<' '<<min<<endl;
		for(int i=1;i<=n;i++){
			if(!vis[i]&&(dis[i]>G[s][i])){
				dis[i]=G[s][i];
			}
		}
	}
	return cost;
}
int main(){
	int a,b,c;
	memset(vis,0,sizeof(vis));
	memset(dis,0x3f3f3f3f,sizeof(dis)); 
	memset(G,0x3f3f3f3f,sizeof(G));//此處應初始化為最大值 
	scanf("%d%d",&n,&m);
	for(int i=0;i<m;i++){
		scanf("%d%d%d",&a,&b,&c);
			G[a][b]=c;
			G[b][a]=c;
	}
	int ans=p();
	cout<<ans<<endl;
	return 0;
}