現有村落間道路的統計數據表中,列出了有可能建設成標準公路的若干條道路的成本,求使每個村落都有公路連通所需要的最低成本。
輸入格式:
輸入數據包括城鎮數目正整數N(≤)和候選道路數目M(≤);隨后的M行對應M條道路,每行給出3個正整數,分別是該條道路直接連通的兩個城鎮的編號以及該道路改建的預算成本。為簡單起見,城鎮從1到N編號。
輸出格式:
輸出村村通需要的最低成本。如果輸入數據不足以保證暢通,則輸出?,表示需要建設更多公路。
輸入樣例:
6 15
1 2 5
1 3 3
1 4 7
1 5 4
1 6 2
2 3 4
2 4 6
2 5 2
2 6 6
3 4 6
3 5 1
3 6 1
4 5 10
4 6 8
5 6 3
輸出樣例:
12
#include<cstdio> #include<cstring> const int maxn = 1010; const int INF = 100000000;int map[maxn][maxn],c[maxn]; bool vis[maxn] = {false}; int n,m;void init(); void prim();int main(){scanf("%d%d",&n,&m);init();int u,v;for(int i = 0; i < m; i++){scanf("%d%d",&u,&v);scanf("%d",&map[u][v]);map[v][u] = map[u][v];}prim();return 0; }void init(){for(int i = 1; i <= n; i++){for(int j = 1; j <= n; j++){if(i == j){map[i][j] = 0;}else{map[i][j] = INF;}}} }void prim(){vis[1] = true;for(int i = 1; i <= n; i++){c[i] = map[i][1];}int sum = 0;bool flag = true;for(int i = 2; i <= n; i++){int u = -1,min = INF;for(int j = 1; j <= n; j++){if(!vis[j] && c[j] < min){u = j;min = c[j]; }}if(u == -1){flag = false;break;}vis[u] = true;sum += min;for(int v = 1; v <= n; v++){if(!vis[v] && c[v] > map[u][v]){c[v] = map[u][v];}}}if(flag) printf("%d",sum);else printf("-1"); }
prim最小生成樹算法---加點法,即選取集合中代價最小的點生成樹