2021牛客寒假算法基础集训营5 E树上博弈 状压dp

原题链接：https://ac.nowcoder.com/acm/contest/9985/E

题意

分析

先点几个细节：

1. 首先这是一个博弈题，双方肯定都是最优策略，因此每一回合双方都是向着对自己最有利的方向去
2. 考虑到数据范围是20，因此状压肯定逃不掉了
3. 注意题中给的是无根树，因此任意度为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();
}