E. Shuffle
题意
给定一颗 n n n 个节点的树 T T T,现在要求恰好执行下面的程序操作一次:
- 从 T T T 中选择一个点,作为 T 2 T_2 T2 的新根节点
- 将这个点从 T T T 中删除,现在 T T T 被分成了一个或一个以上的连通块树
- 对每一个连通块重复上述的操作,并将这个新的点连接到上一个步骤添加到 T 2 T_2 T2 的点
即上述操作是一个递归的过程。
现在要回答经过一次上次的程序后,生成的 T 2 T_2 T2 最多有多少个叶节点
思路
首先我们需要指出结论:除去 T 2 T_2 T2 的根节点,我们能拥有的最大叶子数量就是各个连通块的最大独立集,如果根节点度数为一,我们需要特判并给答案额外加一
这是因为:在我们选择第一个点作为 T 2 T_2 T2 的根节点后,剩下的每一对相邻的点都不可能同时成为叶子,因为它们必定存在一个先后顺序,那么先被选择的那个一定会成为后被选择的那个点的祖先。
那么我们可以得出结论(必要条件):最后的叶子集合一定是一个独立集(点集内部没有边)
假设我们已经有了一个独立集作为最终的目标叶子集合,我们如何保证它们一定是 T 2 T_2 T2 的叶子呢?
我们可以每一步都选择不在 M I S MIS MIS (maximum independent set) 中的点,让它们连接到 T 2 T_2 T2 上,那么最后一定会只剩下我们的叶子集合,即我们的 M I S MIS MIS,它们连接到 T 2 T_2 T2 上自然就是叶子了。
那么到了这里,答案很显然就是最大独立集的大小了,最后特判一下根节点的度数
问题是,我们第一步选择的 T 2 T_2 T2 的根节点是不固定的,我们可以考虑在 T T T 上做 d f s dfs dfs 换根 D P DP DP,通过枚举当前点 u u u 作为根节点,并统计答案
除了当前节点 u u u 外,它将 T T T 分割成了若干个连通子树,那么这些子树内的最大独立集我们可以用类似 树形 D P DP DP 模板题 ”没有上司的舞会“ 的处理方式,以 d p [ u ] [ 0 ] dp[u][0] dp[u][0] 表示 u u u 的子树中,不选 u u u 的最多能选择的点, d p [ u ] [ 1 ] dp[u][1] dp[u][1] 表示选了 u u u,按照独立集的定义,如果 u u u 被选了,那么他的邻居都不能被选
问题在于,如何换根,在线统计当前节点 u u u 的祖先节点的信息?
注意到其实 u u u 的祖先节点也可以看成一颗连通树,我们以 s [ 0 ] s[0] s[0] 和 s [ 1 ] s[1] s[1] 分别表示不选 u u u 的祖先和选了 u u u 的祖先的最大独立集大小。
那么从 u u u 转移到它的儿子 v v v 时, s s s 的变化是: s [ 0 ] = max { s 0 , s 1 } + ∑ e ≠ v ∧ e ∈ s o n { u } max { d p e , 0 , d p e , 1 } s[0] = \max \{ s_0, s_1 \} + \sum_{e \neq v \wedge e \in son \{u \}} \max \{dp_{e, 0}, dp_{e, 1} \} s[0]=max{s0,s1}+e=v∧e∈son{u}∑max{dpe,0,dpe,1}
意思就是:转移到 v v v 时,父亲变为了 u u u,那么 u u u 不选的话, u u u 的其他儿子可以选或不选(取 max \max max), u u u 原本的父亲也可以选或不选
同理, s [ 1 ] = 1 + s 0 + ∑ e ≠ v ∧ e ∈ s o n { u } d p e , 0 s[1] = 1 + s_0 + \sum_{e \neq v \wedge e \in son \{u \}} dp_{e, 0} s[1]=1+s0+e=v∧e∈son{u}∑dpe,0
表示选了 u u u,并且其他邻居都不能选。
时间复杂度: O ( n ) O(n) O(n)
#include<bits/stdc++.h>
#define fore(i,l,r) for(int i=(int)(l);i<(int)(r);++i)
#define fi first
#define se second
#define endl '\n'
#define ull unsigned long long
#define ALL(v) v.begin(), v.end()
#define Debug(x, ed) std::cerr << #x << " = " << x << ed;const int INF=0x3f3f3f3f;
const long long INFLL=1e18;typedef long long ll;const int N = 200005;std::vector<int> g[N];
int dp[N][2];
int sum_mx[N];
int sum0[N];
int ans;void dfs0(int u, int fa){dp[u][1] = 1;dp[u][0] = 0;sum_mx[u] = sum0[u] = 0;for(auto v : g[u])if(v != fa){dfs0(v, u);sum_mx[u] += std::max(dp[v][0], dp[v][1]);sum0[u] += dp[v][0];dp[u][0] += std::max(dp[v][0], dp[v][1]);dp[u][1] += dp[v][0];}
}void dfs1(int u, int fa, std::array<int, 2> s){ans = std::max(ans, sum_mx[u] + std::max(s[0], s[1]) + (g[u].size() == 1));for(auto v : g[u])if(v != fa){std::array<int, 2> a;a[0] = std::max(s[0], s[1]) + sum_mx[u] - std::max(dp[v][0], dp[v][1]);a[1] = 1 + s[0] + sum0[u] - dp[v][0];dfs1(v, u, a);}
}int main(){std::ios::sync_with_stdio(false);std::cin.tie(nullptr);std::cout.tie(nullptr);int t;std::cin >> t;while(t--){int n;std::cin >> n;fore(i, 1, n){int u, v;std::cin >> u >> v;g[u].push_back(v);g[v].push_back(u);}dfs0(1, 0);ans = 0;dfs1(1, 0, {0, 0});std::cout << ans << endl;fore(i, 1, n + 1) g[i].clear();}return 0;
}