2439四叶草魔杖

(clover.cpp/.in/.out)　
　　魔杖护法Freda融合了四件武器，于是魔杖顶端缓缓地生出了一棵四叶草，四片叶子幻发着淡淡的七色光。圣剑护法rainbow取出了一个圆盘，圆盘上镶嵌着N颗宝石，编号为0~N-1。第i颗宝石的能量是Ai。如果Ai>0，表示这颗宝石能量过高，需要把Ai的能量传给其它宝石；如果Ai<0，表示这颗宝石的能量过低，需要从其它宝石处获取-Ai的能量。保证∑Ai =0。只有当所有宝石的能量均相同时，把四叶草魔杖插入圆盘中央，才能开启超自然之界的通道。
不过，只有M对宝石之间可以互相传递能量，其中第i对宝石之间无论传递多少能量，都要花费Ti的代价。探险队员们想知道，最少需要花费多少代价才能使所有宝石的能量都相同?

输入格式

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

输出格式

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

样例输入

3 3
50 -20 -30
0 1 10
1 2 20
0 2 100

样例输出

30

提示

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

题解

dp+状压+最小生成树

将每一个和为零的联通块看做一个商品，然后通过状压dp求出最小值。

代码

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cmath>
#include<queue>
#include<cstring>
using namespace std;
#define maxn1 900
#define maxn2 200000
struct node{
    int a,b,l;
};
int lim;
int n,m;
int a[maxn2],tot[maxn2],v[maxn2],f[maxn2],sum[maxn2];
int dp[maxn2];
node len[maxn1];
int gf(int x)
{
    if(x==f[x]) return x;
    f[x]=gf(f[x]);
    return f[x];
}
int k(int x)
{
    int cnt=0,sum=0,i,j,k,num=0;
    for(i=1;i<=n;i++)
    {
        if((1<<i-1)&x) num++;
    }

    for(i=1;i<=n;i++) f[i]=i;
    for(i=1;i<=m;i++)
    {
        if(((1<<len[i].a-1)&x)&&((1<<len[i].b-1)&x)){
            int tx=gf(len[i].a),ty=gf(len[i].b);
            if(tx==ty) continue;
            f[tx]=ty;
            cnt++;
            sum+=len[i].l;
        }
        if(cnt==num-1) break;
    }
    if(cnt==num-1) return sum;
    else return 1e9;
}
bool cmp(node a,node b)
{
    return a.l<b.l;
}
int main()
{
    int i,j;
    scanf("%d%d",&n,&m);
    for(i=1;i<=n;i++) scanf("%d",&a[i]);
    for(i=1;i<=m;i++) {
    scanf("%d%d%d",&len[i].a,&len[i].b,&len[i].l);
    len[i].a++;len[i].b++;}
    lim=(1<<n)-1;
    sort(len+1,len+1+m,cmp);

    for(i=1;i<=lim;i++){
        for(j=1;j<=n;j++){
            if(i&(1<<(j-1))) tot[i]+=a[j];
        }
    }  
//  for(i=1;i<=lim;i++) cout<<tot[i]<<" ";
    for(i=1;i<=lim;i++){
        if(!tot[i]){
           v[i]=k(i);
        }
        else v[i]=1e9;
    }
    for(i=1;i<=lim;i++) dp[i]=1e9;
    for(i=0;i<=lim;i++)
    {
        if(tot[i]) continue;
        for(j=0;j<=lim;j++)
        {
            if(tot[j]) continue;
            dp[j|i]=min(dp[j|i],dp[j]+v[i]);

        }
    }
    if(dp[lim]==1e9) cout<<"Impossible";
    else cout<<dp[lim];
}
/*3 3
50 -20 -30
0 1 10
1 2 20
0 2 100
7 6
10 -10 10 -10 -20 10 10
0 1 10
1 2 10
2 3 10
2 4 100
4 5 10
5 6 10
*/