安慰奶牛
阿新 • • 發佈:2019-01-25
問題描述:
Farmer John變得非常懶,他不想再繼續維護供奶牛之間供通行的道路。道路被用來連線N個牧場,牧場被連續地編號為1到N。每一個牧場都是一個奶牛的家。FJ計劃除去P條道路中儘可能多的道路,但是還要保持牧場之間 的連通性。你首先要決定那些道路是需要保留的N-1條道路。第j條雙向道路連線了牧場Sj和Ej(1 <= Sj <= N; 1 <= Ej <= N; Sj != Ej),而且走完它需要Lj的時間。沒有兩個牧場是被一條以上的道路所連線。奶牛們非常傷心,因為她們的交通系統被削減了。你需要到每一個奶牛的住處去安慰她們。每次你到達第i個牧場的時候(即使你已經到過),你必須花去Ci
第1行包含兩個整數N和P。
接下來N行,每行包含一個整數Ci。
接下來P行,每行包含三個整數Sj, Ej和Lj。
輸出格式 輸出一個整數, 所需要的總時間(包含和在你所在的牧場的奶牛的兩次談話時間)。 樣例輸入 5 710
10
20
6
30
1 2 5
2 3 5
2 4 12
3 4 17
2 5 15
3 5 6
4 5 12 解題思路: 1:因為結點數量較大,prim演算法勢必要超時,故選用kruskl. 2:每次走過一個結點,都要停留Ci時間,實際中走的時候,在最小生成樹走的時候實際上要走倆次,因此邊權為Lj*2+C[s]+C[e];題目中要求最後要回到原點,所以還要加一個出發點的權值,出發點當然一定要選最小的!
#include<cstdio> #include<cstring> #include<iostream> #include<algorithm> using namespace std; struct Edge { int x; int y; int cost; }edge[100001]; int set[100001]; int val[100001]; inline int find(int x) { int i,j,r; r=x; while(set[r]!=r) r=set[r]; i=x; while(i!=r) { j=set[i]; set[i]=r; i=j; } return r; } inline void merge(int x,int y) { int t=find(x); int h=find(y); if(t<h) set[h]=t; else set[t]=h; } bool cmp(const Edge& a,const Edge& b) { return a.cost<b.cost; } void init(int n) { for(int i=1;i<=n;i++) set[i]=i; } int kruskal(int n,int m) { int i,num,sum,from,to; sort(edge,edge+m,cmp); sum=num=0; for(i=0;i<m;i++) { from = find(edge[i].x); to = find(edge[i].y); if(from != to) { merge(from,to); sum+=edge[i].cost; num++; } if(num==n-1) break; } if(num<n-1) return -1; else return sum; } int main() { int i,n,m,k; int minn=0x3f3f3f3f; memset(val,0,sizeof(val)); while(scanf("%d%d",&n,&m)!=EOF) { init(n); for(i=1;i<=n;i++) { scanf("%d",&val[i]); minn=min(minn,val[i]); } for(i=0;i<m;i++) { scanf("%d%d%d",&edge[i].x,&edge[i].y,&edge[i].cost); edge[i].cost+=edge[i].cost; edge[i].cost+=val[edge[i].x]+val[edge[i].y]; } k=kruskal(n,m)+minn; if(k==-1) printf("no\n"); else printf("%d\n",k); } }