1. 程式人生 > >安慰奶牛

安慰奶牛

問題描述:

Farmer John變得非常懶,他不想再繼續維護供奶牛之間供通行的道路。道路被用來連線N個牧場,牧場被連續地編號為1到N。每一個牧場都是一個奶牛的家。FJ計劃除去P條道路中儘可能多的道路,但是還要保持牧場之間 的連通性。你首先要決定那些道路是需要保留的N-1條道路。第j條雙向道路連線了牧場Sj和Ej(1 <= Sj <= N; 1 <= Ej <= N; Sj != Ej),而且走完它需要Lj的時間。沒有兩個牧場是被一條以上的道路所連線。奶牛們非常傷心,因為她們的交通系統被削減了。你需要到每一個奶牛的住處去安慰她們。每次你到達第i個牧場的時候(即使你已經到過),你必須花去Ci

的時間和奶牛交談。你每個晚上都會在同一個牧場(這是供你選擇的)過夜,直到奶牛們都從悲傷中緩過神來。在早上 起來和晚上回去睡覺的時候,你都需要和在你睡覺的牧場的奶牛交談一次。這樣你才能完成你的 交談任務。假設Farmer John採納了你的建議,請計算出使所有奶牛都被安慰的最少時間。

輸入格式

第1行包含兩個整數N和P。

接下來N行,每行包含一個整數Ci

接下來P行,每行包含三個整數Sj, Ej和Lj

輸出格式 輸出一個整數, 所需要的總時間(包含和在你所在的牧場的奶牛的兩次談話時間)。 樣例輸入 5 7
10
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);
	}
}