(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
*/