bzoj4871 [Shoi2017]摧毁“树状图”(树形dp)

求用两条边不相交路径最多可以把一棵树分成几个连通块。
细节繁多的树形dp囧
orz ltx 这个人有图呢!portal

大概就是讨论一下不同的链形态怎么转移qaq

#include 
using namespace std;
#define inf 0x3f3f3f3f
#define ll long long
#define inf 0x3f3f3f3f
#define N 500010
inline char gc(){
    static char buf[1<<16],*S,*T;
    if(T==S){T=(S=buf)+fread(buf,1,1<<16,stdin);if(T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(ch<'0'||ch>'9'){if(ch=='-') f=-1;ch=gc();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=gc();
    return x*f;
}
int tst,ty,n,h[N],num,du[N],ans,dp[N][4];
struct edge{
    int to,next;
}data[N<<1];
inline void Max(int &x,int y){x=max(x,y);}
void dfs(int x,int Fa){
    dp[x][0]=dp[x][2]=dp[x][3]=du[x];dp[x][1]=1;int tmp=0;
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(y==Fa) continue;dfs(y,x);
        Max(ans,dp[x][0]+dp[y][3]-1+(x!=1));
        Max(ans,dp[x][3]+dp[y][0]-1+(x!=1));
        Max(ans,dp[x][1]+dp[y][1]-1);
        Max(ans,dp[x][1]+dp[y][2]);
        Max(ans,dp[x][2]+dp[y][1]-1+(x!=1));
        Max(ans,dp[x][2]+dp[y][2]-1+(x!=1));
        Max(dp[x][3],dp[y][3]+du[x]-1);
        Max(dp[x][3],dp[y][1]+dp[x][0]-1);
        Max(dp[x][3],dp[y][2]+dp[x][0]-1);
        Max(dp[x][3],dp[y][0]+tmp-2+du[x]);
        Max(dp[x][3],dp[y][0]+dp[x][2]-1);
        Max(dp[x][2],dp[x][0]+dp[y][0]-1);
        Max(dp[x][1],dp[y][1]);
        Max(dp[x][1],dp[y][2]+1);
        Max(dp[x][0],dp[y][0]+du[x]-1);
        Max(dp[x][2],dp[x][0]);
        Max(dp[x][3],dp[x][2]);
        Max(tmp,dp[y][1]);Max(tmp,dp[y][2]);
    }du[x]=h[x]=0;
}
int main(){
//  freopen("a.in","r",stdin);
    tst=read();ty=read();
    while(tst--){
        n=read();for(int i=1;i<=ty;++i) read(),read();num=0;ans=0;
        for(int i=2;i<=n;++i){
            int x=read(),y=read();du[i]--;
            data[++num].to=y;data[num].next=h[x];h[x]=num;du[y]++;
            data[++num].to=x;data[num].next=h[y];h[y]=num;du[x]++;
        }dfs(1,0);
        printf("%d\n",ans);
    }return 0;
}

你可能感兴趣的:(bzoj,树形dp)