打卡信奥刷题（3522）用C++实现信奥题 P10949 四叶草魔杖

原创于 2026-08-22 07:13:02 发布 · 277 阅读

3 ·

本内容遵循CC 4.0 BY-SA版权协议

GEO检测

收录于

C++

P10949 四叶草魔杖

题目描述

魔杖护法 Freda 融合了四件武器，于是魔杖顶端缓缓地生出了一棵四叶草，四片叶子幻发着淡淡的七色光。

圣剑护法 rainbow 取出了一个圆盘，圆盘上镶嵌着 $N$ 颗宝石，编号为 $\sim N-1$ 。第 $i$ 颗宝石的能量是 $A_i$ 。

如果 $A_i > 0$ ，表示这颗宝石能量过高，需要把 $A_i$ 的能量传给其它宝石；如果 $A_i < 0$ ，表示这颗宝石的能量过低，需要从其它宝石处获取 $A_i$ 的能量。

保证 $∑Ai=0\sum A_i = 0$ 。

只有当所有宝石的能量均相同时，把四叶草魔杖插入圆盘中央，才能开启超自然之界的通道。

不过，只有 $M$ 对宝石之间可以互相传递能量，其中第 $i$ 对宝石之间无论传递多少能量，都要花费 $T_i$ 的代价。

探险队员们想知道，最少需要花费多少代价才能使所有宝石的能量都相同？

输入格式

第一行两个整数 $N, M$ 。

第二行 $N$ 个整数 $A_i$ 。

接下来 $M$ 行每行三个整数 $p_i,q_i,T_i$ ，表示在编号为 $p_i$ 和 $q_i$ 的宝石之间传递能量需要花费 $T_i$ 的代价。

数据保证每对 $p_i,q_i$ 最多出现一次。

输出格式

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

输入输出样例 #1

输入 #1

输出 #1

说明/提示

$\le N \le 16$ ,
$\le M \le \dfrac{N(N-1)}{2}$ ,
$\le p_i,q_i < N$ ,
$\le A_i \le 1000$ ,
$\le T_i \le 1000$

C++实现

#include <iostream>
#include <cstring>
#include <algorithm>
#define reg register
using namespace std;
const int N=17;
int n,m,dp[1<<N],sum[1<<N],mst[1<<N],b[1<<N];
struct edge{
	int u,v,w;
} e[300];
int main(){
	ios::sync_with_stdio(0);
	cin>>n>>m;reg int x,y,z;
	for (int i=0;i<n;++i)cin>>sum[1<<i];
	for (int i=0;i<m;++i)cin>>e[i].u>>e[i].v>>e[i].w;
	for (int i=2;i<1<<n;++i)sum[i]=sum[i^(i&-i)]+sum[i&-i];
	memset(mst,0x3f,sizeof mst);mst[0]=0;
	for (int i=0;i<n;++i)mst[1<<i]=0;
	for (int i=1;i<1<<n;++i)
		for (int j=0;j<m;++j){
			x=e[j].u;y=e[j].v;z=e[j].w;
			if (!(i&(1<<x))||!(i&(1<<y)))continue; 
			mst[i]=min({mst[i],mst[i^(1<<x)]+z,mst[i^(1<<y)]+z});
		}
	memset(dp,0x3f,sizeof dp);dp[0]=0;
	for (reg int i=1;i<1<<n;++i){
		if (sum[i])continue;
		for (x=0,y=i;y;++x,y^=y&-y)b[1<<x]=y&-y;
		for (reg int j=0;j<1<<x;++j){
			for (y=j,z=0;y;y^=y&-y)z|=b[y&-y];
			if (sum[z])continue;
			dp[i]=min(dp[i],dp[z]+mst[i^z]);
		}
	}
	if (dp[(1<<n)-1]>=0x3f3f3f3f)
		cout<<"Impossible";
	else 
		cout<<dp[(1<<n)-1];
	return 0;
}