上海计算机学会 2023年11月月赛 乙组T3 树的匹配（树形DP、乘法逆元）

第三题：T3树的匹配

标签：树形$DP$、乘法逆元
题意：给定一棵有$n$个节点的树，根节点为$1$。求这棵树的最大匹配数，并统计最大匹配数情况下的方案数。最终结果，对$10^9+7$取余。树的匹配，指的是具有父子关系的点，两两组成一对，每个点只能在一个配对里。
题解：很明显这是一道树形动态规划。
$dp[u][0/1]$：节点编号为$u$的子树中，$u$不配对/配对时其子树的最大匹配数。
$dp[u][0]={\sum }_{v}Max(dp[v][0],dp[v][1])$
当$u$不配对的时候，把所有孩子节点$v$最大匹配数（$v$不配对或者配对的情况下中的最大值）加起来。
统计完$dp[u][0]$之后，$dp[u][1]$表示的是配对的情况，需要从所有孩子节点$v$没有配对的情况转移过来；当前的孩子节点$v^{\prime }$配对的情况下，其他的孩子节点还是拿$max(dp[v][0],dp[v][1])$。
$dp[u][1]=Ma{x}_j(dp[u][0]-max(dp[v][0],dp[v][1])+dp[v][0]+1)$
这边的$+1$指的是$u$和当前枚举的孩子节点组成的一个新的匹配数。
可以根据下面的图，同学自己再推一推。

$cnt[u][0/1]$：节点编号为$u$的子树中，$u$不配对/配对时其子树的最大匹配数下的方案数
$cnt[u][0]$其实比较好想，$u$节点既然不配对了，那么只需要管它的所有孩子节点$v$，我们是求最大匹配数的情况下方案数，所以我们得比较一下$dp[v][0]$和$dp[v][1]$中哪个传上来的最大匹配数大，或者一样大。
所以这边我写了一个$select(v)$函数：
$select(v)=dp[v][0]>=dp[v][1]\ast cnt[v][0]+dp[v][0]<=dp[v][1]\ast cnt[v][1]$

ll select(ll v) {
    if (dp[v][0] > dp[v][1]) return cnt[v][0];
    else if (dp[v][0] < dp[v][1]) return cnt[v][1];
    else return (cnt[v][0] + cnt[v][1]) % mod;
}

接下来考虑一下，对于$u$节点来说，所有子树的方案数传上来，是不是得乘积一下（乘法原理）。
公式化：$cnt[u][0]={\prod }_v(select(v))$
然后这边我们从上面的$dp[u][1]$的情况，推一推，上面的是加法原理，然后$u$和某一个孩子节点$v$进行匹配的情况下，我们是先减去$max(dp[v][0],dp[v][1])$，那同理 这边乘法原理，我们要去求$cnt[u][1]$是不是得先除一下。

那接下来我们看实际在什么情况下进行转移，如果在$dp[u][1]$能够变的更大的时候，当然直接从$cnt[v][0]$（$v$不选的情况）的地方转移过来。即$cnt[u][1]=cnt[u][0]/select(v)\ast cnt[v][0]$

那如果当前和之前$dp[u][1]$的值一样，那也就是说有全新的也能够组成最大匹配数的情况，是不是得求和加上来。即$cnt[u][1]=cnt[u][1]+cnt[u][0]/select(v)\ast cnt[v][0]$

需要额外考虑的点是，整道题的方案数最终结果需要取模，除法运算是不符合同余定理的，我们得使用逆元给它转一下。这边通过快速幂去实现一下费马小定理求逆元的。

费马小定理：如果$p$是一个质数，而整数$a$不是$p$的倍数，则有$a^{p-1}\equiv 1(modp)$。
推导得$a^{p-2}=inv(a)(modp)$=> $inv(a)=a^{p-2}(modp)$
代码：

#include <bits/stdc++.h>
using namespace std;

typedef long long ll;
const ll mod = 1e9 + 7;

ll n, x;
ll dp[200005][2], cnt[200005][2];
// dp[i][0/1]: i这个顶点不放/放到最终匹配方案中
vector <ll> e[200005];

ll fast_power(ll a, ll b) {
    ll ans = 1;
    a %= mod;
    while (b) {
        if (b & 1) ans = (ans * a) % mod;
        a = (a * a) % mod;
        b >>= 1;
    }
    return ans;
}

ll select(ll v) {
    if (dp[v][0] > dp[v][1]) return cnt[v][0];
    else if (dp[v][0] < dp[v][1]) return cnt[v][1];
    else return (cnt[v][0] + cnt[v][1]) % mod;
}

void dfs(ll u) {
    cnt[u][0] = 1;
    for (int i = 0; i < e[u].size(); i++) {
        ll v = e[u][i];
        dfs(v);
        dp[u][0] += max(dp[v][0], dp[v][1]);
        cnt[u][0] = (cnt[u][0] * select(v)) % mod;
    }
    for (int i = 0; i < e[u].size(); i++) {
        ll v = e[u][i];
        ll otrher = dp[u][0] - max(dp[v][0], dp[v][1]);
        if (dp[u][1] < dp[v][0] + otrher + 1) {
            dp[u][1] = dp[v][0] + otrher + 1;
            ll k = cnt[u][0] * fast_power(select(v), mod - 2) % mod;
            cnt[u][1] = (k * cnt[v][0] + mod) % mod;
        }
        else if (dp[u][1] == dp[v][0] + otrher + 1) {
            ll k = cnt[u][0] * fast_power(select(v), mod - 2) % mod;
            cnt[u][1] = (cnt[u][1] + (k * cnt[v][0]) % mod + mod) % mod;
        }
    }
}

int main() {
    cin >> n;
    for (int i = 2; i <= n; i++) {
        cin >> x;
        e[x].push_back(i);
    }
    dfs(1);
    cout << max(dp[1][0], dp[1][1]) << endl;
    cout << select(1) << endl;
    return 0;
}