原题链接:https://ac.nowcoder.com/acm/contest/9985/E
题意
分析
先点几个细节:
- 首先这是一个博弈题,双方肯定都是最优策略,因此每一回合双方都是向着对自己最有利的方向去
- 考虑到数据范围是20,因此状压肯定逃不掉了
- 注意题中给的是无根树,因此任意度为1的点都可以被删掉
然后开始考虑状压的含义,如果我们分两个人考虑,先手加上得分取max,后手减去得分取min,这样理论上是可以的,但实现起来略显复杂,因此有什么方法可以将状态对于两个人都适用?可以设dp[state]表示当前取完后与前者分差最优的值,这样我们每次转移时都维护max就可以了,状态转移方程即为f[i | (1<<j)] = max(f[i | (1 << j)], a[j] - dp[i])。举个例子,当前点的权值为3,你在取之前比前者多4分,那么前者的状态就是-4, 转移过后就是7,也就是你领先了7。
接着我们如何快速找到加入进来的点,因为加进来的必须是叶子节点,因此提前预处理一下每个点连的边g[u] = 1 << v,这样处理完之后,如果g[u] & i不为0说明这个点与集合是有边的,而且树形结构不用担心会形成环,所以直接加就可以啦。
Code
#include <bits/stdc++.h>
using namespace std;
//#define ACM_LOCAL
#define fi first
#define se second
#define il inline
#define re register
typedef long long ll;
typedef pair<int, int> PII;
typedef unsigned long long ull;
const int N = 1e5 + 10;
const int M = 1e6 + 10;
const int INF = 1e9 + 5;
const double eps = 1e-5;
const int MOD = 998244353;
ll dp[1 << 21];
void solve() {
int T; cin >> T; while (T--) {
int n; cin >> n;
memset(dp, 128, sizeof dp);
vector<int> a(n+1), g(n+1);
for (int i = 0; i < n; i++) cin >> a[i], g[i] = 0;
for (int i = 1; i <= n-1; i++) {
int u, v; cin >> u >> v;
u--, v--;
g[u] |= 1 << v;
g[v] |= 1 << u;
}
int cnt = 1 << n;
for (int i = 0; i < n; i++) dp[1 << i] = a[i];
for (int i = 0; i < cnt; i++) {
if (dp[i] < -1e12) continue;
for (int j = 0; j < n; j++) {
if (!(i >> j & 1) && (g[j] & i)) {
dp[i | (1 << j)] = max(dp[i | (1 << j)], a[j] - dp[i]);
}
}
}
cout << dp[(1 << n) - 1] << endl;
}
}
signed main() {
ios::sync_with_stdio(0);
cin.tie(0);
cout.tie(0);
#ifdef ACM_LOCAL
freopen("input", "r", stdin);
freopen("output", "w", stdout);
#endif
solve();
}