BSOJ3809 tyvj2054 [Nescafé29]四叶草魔杖 神级骗分 数据之水

3809 -- 【模拟试题】四叶草魔杖

Description

　　魔杖护法Freda融合了四件武器，于是魔杖顶端缓缓地生 出了一棵四叶草，四片叶子幻发着淡淡的七色光。圣剑护法rainbow取出了一个圆盘，圆盘上镶嵌着N颗宝石，编号为0~N-1。第i颗宝石的能量是 Ai。如果Ai>0，表示这颗宝石能量过高，需要把Ai的能量传给其它宝石；如果Ai<0，表示这颗宝石的能量过低，需要从其它宝石处获取 -Ai的能量。保证∑Ai=0。只有当所有宝石的能量均相同时，把四叶草魔杖插入圆盘中央，才能开启超自然之界的通道。

　　不过，只有M对宝石之间可以互相传递能量，其中第i对宝石之间无论传递多少能量，都要花费Ti的代价。探险队员们想知道，最少需要花费多少代价才能使所有宝石的能量都相同？

Input

　　第一行两个整数N、M。

　　第二行N 个整数Ai。

　　接下来M行每行三个整数pi,qi,Ti，表示在编号为pi和qi的宝石之间传递能量需要花费Ti的代价。数据保证每对pi、qi最多出现一次。

Output

　　输出一个整数表示答案。无解输出Impossible。

Sample Input

3 3

50 -20 -30

0 1 10

1 2 20

0 2 100

Sample Output

30

Hint

【数据规模】

　　 对于50%的数据，2<=N<=8。

　　对于100%的数据，2<=N<=16，0<=M<=N*(N-1)/2，0<=pi,qi<=ai<=1000，

0<=Ti<=1000，∑Ai=0。

用最小生成树，然后累加边权和。之前判断一下无解就可以了。数据太水了，可能是随机生成的吧……

#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<cstdio>
using namespace std;
struct node{int fr,to,val;
}edge[10005];
int fa[10005],n,m,v[10005],du[10005];
bool cmp(node a,node b)
{
	return a.val<b.val;
}
int G(int x)
{return fa[x]==x?x:fa[x]=G(fa[x]);}
void K()
{
	sort(edge+1,edge+m+1,cmp);
	int ans=0,cnt=0;
	for(int i=1;i<=n;i++)
	  fa[i]=i;
	
	for(int i=1;i<=m;i++)
	{
		int fa1=G(edge[i].fr);
		int fa2=G(edge[i].to);
		if(fa1!=fa2)
		{
			fa[fa1]=fa2;
			ans+=edge[i].val;
			cnt++;
		}
		if(cnt==n-1) break;
	}
	cout<<ans;
}
int main(){
	cin>>n>>m;
	for(int i=1;i<=n;i++)cin>>v[i];
	for(int i=1;i<=m;i++)cin>>edge[i].fr>>edge[i].to>>edge[i].val,du[edge[i].fr+1]++,du[edge[i].to+1]++;
	for(int i=1;i<=n;i++)
	  if(du[i]==0&&v[i]){cout<<"Impossible";return 0;}
	K();
	return 0;
}