脚本宝典收集整理的这篇文章主要介绍了CF1442E Black, White and Grey Tree 题解,脚本宝典觉得挺不错的,现在分享给大家,也给大家做个参考。
Codeforces Luogu
给定一个树,每个点颜色可能是黑、白、灰。 你每次选择一个联通块,在里面选择若干个点,然后删去它和它们的边。 同时,你不能同时选择白点和黑点,问至少多少次删完。
首先考虑没有灰点的问题。 设一条边的权值 (w=[c_une c_v]),答案就是 (leftlceilfrac {len}2rightrceil+1),其中 (len) 表示带权直径。
然后考虑灰点可以被当作任意颜色。 就可以直接树型 dp 求最短直径。 考虑设 (f_{u,0/1}) 表示 (u) 为根到叶子的最长路。 (g_{u,0/1}) 表示 (u) 为根的路径最小值。 答案就是 (max_{u}(min(g_{u,0},g_{u,1})))。 直接转移就行了? 注意最大值最小值要搞清楚
//Coded by Kamiyama_Shiki on 2021.11.08 {{{
//是啊……你就是那只鬼了……所以被你碰到以后,就轮到我变成鬼了
#include<bits/stdc++.h>
using namespace std;typedef long long ll;
template<typename T>inline void read(T &x)
{
x=0;char c=getchar(),f=0;
for(;c<48||c>57;c=getchar()) if(!(c^45)) f=1;
for(;c>=48&&c<=57;c=getchar()) x=(x<<1)+(x<<3)+(c^48);
f?x=-x:x;
}
template<typename T,typename...L>inline void read(T &x,L&...l) {read(x),read(l...);}//}}}
const int N=200005;int n,a[N],rs=0,f[N][2],g[N][2];
struct edge{int to,nxt;}e[N<<1];int et,head[N];
inline void adde(int x,int y) {e[++et]=(edge){y,head[x]},head[x]=et;}
inline void dfs(int x,int fa)
{
if(a[x]==3) g[x][0]=g[x][1]=-1e7,f[x][0]=f[x][1]=0;
else g[x][a[x]&1]=-1e7,f[x][a[x]&1]=0,g[x][(a[x]&1)^1]=f[x][(a[x]&1)^1]=1e7;
for(int i=head[x];i;i=e[i].nxt) if(e[i].to!=fa)
{
int y=e[i].to,vl;dfs(y,x);
if(a[x]&2) vl=min(f[y][0],f[y][1]+1),g[x][0]=max(g[x][0],f[x][0]+vl),f[x][0]=max(f[x][0],vl);
if(a[x]&1) vl=min(f[y][1],f[y][0]+1),g[x][1]=max(g[x][1],f[x][1]+vl),f[x][1]=max(f[x][1],vl);
}rs=max(rs,min(g[x][0],g[x][1]));
//printf("%d %d ( %d %d )n",g[x][0],g[x][1],f[x][0],f[x][1]);
}
inline void solve()
{
read(n),rs=0,et=0;for(int i=1;i<=n;i++) head[i]=0;
for(int i=1;i<=n;i++) read(a[i]),a[i]=3^a[i];
for(int i=1,x,y;i<n;i++) read(x,y),adde(x,y),adde(y,x);
dfs(1,0),printf("%dn",(rs+1)/2+1);
}
int main() {int Ca;for(read(Ca);Ca--;) solve();return 0;}
以上是脚本宝典为你收集整理的CF1442E Black, White and Grey Tree 题解全部内容,希望文章能够帮你解决CF1442E Black, White and Grey Tree 题解所遇到的问题。
本图文内容来源于网友网络收集整理提供,作为学习参考使用,版权属于原作者。
如您有任何意见或建议可联系处理。小编QQ:384754419,请注明来意。