JZOJ3392-四叶草魔杖

Description&Data Constraint

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

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

$2\le N\le 16,0\le M\le \frac{N\times (N-1)}{2},0\le p_i,q_i\le N,-10^3\le A_i\le 10^3,0\le T_i\le 10^3,\sum A_i=0$

Solution

由于$N$很小，可以在$O(N\times 2^N)$的时间内把所有联通的且点权和为0的集合找出来，每个集合跑一遍最小生成树

然后考虑状压DP，设$f_s$表示$s$的点权和为0的最小代价，那么部分初值就是找出来的为0集合的最小生成树

然后暴力枚举两个子集$i$、$j$，如果$i$、$j$没有交集，那么$f_{i|j}=f_i+f_j$

Code

#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 20
#define inf 0X3f3f3f3f
using namespace std;
struct node
{
	int x,y,v;
}edge[N*N];
int n,m,x,y,z,xx,yy,ans,res,tot,a[N],f[N],sum[1<<N][N],MAP[N][N],RES[N],dp[1<<N];
bool b[N],bz[N],bj[N];
bool cmp(node x,node y) {return x.v<y.v;}
int find(int x)
{
	if (f[x]!=x) f[x]=find(f[x]);
	return f[x];
}
void kruskal(int x)
{
	int num=0;
	for (int i=1;i<=sum[x][0];++i)
		for (int j=i+1;j<=sum[x][0];++j)
		{
			if (MAP[sum[x][i]][sum[x][j]])
			{
				edge[++num].x=sum[x][i];
				edge[num].y=sum[x][j];
				edge[num].v=MAP[sum[x][i]][sum[x][j]];	
			}	
		}
	for (int i=1;i<=n;++i) f[i]=i;	
	sort(edge+1,edge+num+1,cmp);
	memset(b,true,sizeof(b));
	RES[x]=0;
	for (int i=1;i<=num;++i)
	{
		int X=edge[i].x,Y=edge[i].y,Z=edge[i].v;
		if (!b[X]||!b[Y]) continue;
		xx=find(X);yy=find(Y);
		if (xx!=yy)
		{
			f[xx]=yy;
			RES[x]+=Z;
		}
	}
}
void dfs(int x)
{
	++res;
	bz[x]=true;
	for (int i=1;i<=n;++i)
	{
		if (!bj[i]) continue;
		if (bz[i]) continue;
		if (MAP[x][i]) dfs(i);
	}
}
void dg(int x,int Sum,bool flag)
{
	if (x>n)
	{
		if (!flag) return;
		if (Sum==0)
		{
			res=0;
			int pd0=0,pd[N];
			memset(bz,false,sizeof(bz));
			memset(pd,0,sizeof(pd));
			for (int i=1;i<=n;++i)
				if (bj[i]) pd[++pd0]=i;
			dfs(pd[1]);
			if (res<pd0) return;
			++tot;
			for (int i=1;i<=n;++i)
			{
				if (bj[i])
				{
					++sum[tot][0];
					sum[tot][sum[tot][0]]=i;
				}
			}
			kruskal(tot);
		}
		return;
	}
	bj[x]=true;
	dg(x+1,Sum+a[x],true);
	bj[x]=false;
	dg(x+1,Sum,flag);
}
int main()
{
	scanf("%d%d",&n,&m);
	for (int i=1;i<=n;++i)
		scanf("%d",&a[i]);
	for (int i=1;i<=m;++i)
	{
		scanf("%d%d%d",&x,&y,&z);
		++x;++y;
		MAP[x][y]=MAP[y][x]=z;
	}
	dg(1,0,false);
	memset(dp,inf,sizeof(dp));
	for (int i=1;i<=tot;++i)
	{
		int ss=0;
		for (int j=1;j<=sum[i][0];++j)
			ss+=(1<<(sum[i][j]-1));
		dp[ss]=RES[i];
	}
	for (int i=0;i<(1<<n);++i)
	{
		if (dp[i]==inf) continue;
		for (int j=0;j<(1<<n);++j)
		{
			if (dp[j]==inf) continue;
			if ((i&j)==0) dp[i|j]=min(dp[i|j],dp[i]+dp[j]);
		}
	}
	if (dp[(1<<n)-1]==inf) printf("Impossible\n");
	else printf("%d\n",dp[(1<<n)-1]);
	return 0;
}