题外话:这是魔禁里的那个树状图么
题内话:建议大家找个有图的题解
很容易看出来是树形DP,设计状态:
f
[
x
]
[
0
]
f[x][0]
f[x][0]:穿过x向上的半条链
f
[
x
]
[
1
]
f[x][1]
f[x][1]:不穿过x且完全在子树内的一条链
f
[
x
]
[
2
]
f[x][2]
f[x][2]:穿过x且完全在子树内的一条链
f
[
x
]
[
3
]
f[x][3]
f[x][3]:穿过x向上的半条链以及完全在子树内的一条链
Code:
#include<bits/stdc++.h>
using namespace std;
inline int read(){
int res=0,f=1;char ch=getchar();
while(!isdigit(ch)) {if(ch=='-') f=-f;ch=getchar();}
while(isdigit(ch)) {res=(res<<1)+(res<<3)+(ch^48);ch=getchar();}
return res*f;
}
const int N=5e5+5;
vector<int>e[N];
inline void add(int u,int v){e[u].push_back(v);}
int f[N][4],ans;
int in[N];
inline void upd(int &a,const int &b){a=max(a,b);}
void dfs(int v,int fa) {
bool isrt=v==1;
int deg=e[v].size()-!isrt;
f[v][0]=f[v][2]=f[v][3]=deg;f[v][1]=1;
int maxn=0;
for(int i=0;i<e[v].size();i++) {
int y=e[v][i];
if(y==fa) continue;
dfs(y,v);
upd(ans,f[v][0]+f[y][3]-isrt);
upd(ans,f[v][3]+f[y][0]-isrt);
upd(ans,f[v][1]+f[y][1]-1);
upd(ans,f[v][1]+f[y][2]);
upd(ans,f[v][2]+f[y][1]-isrt);
upd(ans,f[v][2]+f[y][2]-isrt);
upd(f[v][1],f[y][1]);
upd(f[v][1],f[y][2]+1);
upd(f[v][3],f[y][3]+deg-1);
upd(f[v][3],f[y][0]+maxn+deg-2);
upd(f[v][3],f[v][0]+f[y][1]-1);
upd(f[v][3],f[v][0]+f[y][2]-1);
upd(f[v][3],f[v][2]+f[y][0]-1);
upd(f[v][2],f[v][0]+f[y][0]-1);
upd(f[v][0],f[y][0]+deg-1);
upd(f[v][2],f[v][0]);
upd(f[v][3],f[v][2]);
upd(maxn,f[y][1]);
upd(maxn,f[y][2]);
}
}
int main(){
int t=read(),x=read();
while(t--){
int n=read();
for(int i=0;i<x*2;i++) read();
for(int x,y,i=1;i<n;i++){x=read(),y=read();add(x,y);add(y,x);in[x]++,in[y]++;}
ans=0;
dfs(1,0);
cout<<ans<<"\n";
for(int i=1;i<=n;i++){e[i].clear();}
}
return 0;
}