hdu 4714 Tree2cycle dp

用树形dp做的，dp[t][i]表示t及其孩子入度都已经小于等于2并且t这个节点的入度等于i的最优解。

那么转移什么的自己想想就能明白了。

关键在于这个题目会暴栈，所以我用了一次bfs搜索出节点的顺序，然后再计算。

 

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int maxn=1e6+9;
struct
{
    int to,next;
}e[maxn*2];
int head[maxn],lon;
bool text[maxn];
void edgeini()
{
    memset(head,-1,sizeof(head));
    lon=-1;
}
void edgemake(int from,int to)
{
    e[++lon].to=to;
    e[lon].next=head[from];
    head[from]=lon;
}
int dp[maxn][3];
int from[maxn];
void dfs(int t)
{
    int k;
    int a1=10,a2=10,ret=0,u;
    for(k=head[t];k!=-1;k=e[k].next)
    {
        u=e[k].to;
        if(u==from[t]) continue;
        ret++;
        dp[t][0]+=dp[u][2]+2;
        if(dp[u][1]-dp[u][2]<a1)
        {
            a2=a1;
            a1=dp[u][1]-dp[u][2];
        }
        else if(dp[u][1]-dp[u][2]<a2)
        a2=dp[u][1]-dp[u][2];
    }
    if(ret>=1)
    dp[t][1]=max(0,dp[t][0]+a1-2);
    else
    dp[t][1]=dp[t][0];
    if(ret>=2)
    dp[t][2]=max(0,dp[t][0]+a1+a2-4);
    else
    dp[t][2]=dp[t][1];
//    cout<<0<<endl;
}

int que[maxn];
void bfs()
{
    int front=1,end=0;
    que[++end]=1;
    text[1]=1;
    while(front<=end)
    {
        int t=que[front++];
        for(int k=head[t];k!=-1;k=e[k].next)
        {
            int u=e[k].to;
            if(!text[u])
            {
                from[u]=t;
                text[u]=1;
                que[++end]=u;
            }
        }
    }
    for(int i=end;i>=1;i--)
    dfs(que[i]);
}


int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        memset(dp,0,sizeof(dp));
        edgeini();
        int n;
        scanf("%d",&n);
        for(int i=1,from,to;i<n;i++)
        {
            scanf("%d %d",&from,&to);
            edgemake(from,to);
            edgemake(to,from);
        }
        memset(text,0,sizeof(text));
        bfs();
        printf("%d\n",dp[1][2]+1);
    }
    return 0;
}