[BZOJ4871][树形DP]SHOI2017：摧毁“树状图”

BZOJ4871

题外话：这是魔禁里的那个树状图么
题内话：建议大家找个有图的题解

很容易看出来是树形DP，设计状态：
$f[x][0]$：穿过x向上的半条链
$f[x][1]$：不穿过x且完全在子树内的一条链
$f[x][2]$：穿过x且完全在子树内的一条链
$f[x][3]$：穿过x向上的半条链以及完全在子树内的一条链

Code：

#include
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;
}