bzoj 4813: [Cqoi2017]小Q的棋盘【树形dp】

这么简单的dp我怎么没想到x2
f为从这个点出发后回到这个点最多能走过的点，g为从这个点出发后不回到这个点最多能走过的点，注意g有两种转移：g[u][k]=max(g[u][k],f[u][k-j-1]+g[e[i].to][j])是在e[i].to这个子树前走了一棵子树再回来，g[u][k]=max(g[u][k],g[u][k-j-2]+f[e[i].to][j])是走了e[i].to的一棵子树之后回到e[i].to再走另一棵

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int N=105;
int n,m,h[N],cnt,f[N][N],g[N][N],ans;
struct qwe
{
    int ne,to;
}e[N<<1];
void add(int u,int v)
{
    cnt++;
    e[cnt].ne=h[u];
    e[cnt].to=v;
    h[u]=cnt;
}
void dp(int u,int fa)
{
    f[u][0]=g[u][0]=1;
    for(int i=h[u];i;i=e[i].ne)
        if(e[i].to!=fa)
        {
            dp(e[i].to,u);
            for(int k=m;k>=1;k--)
                for(int j=0;j<k;j++)
                {
                    if(k-j-2>=0)
                        f[u][k]=max(f[u][k],f[u][k-j-2]+f[e[i].to][j]);
                    g[u][k]=max(g[u][k],f[u][k-j-1]+g[e[i].to][j]);
                    if(k-j-2>=0)
                        g[u][k]=max(g[u][k],g[u][k-j-2]+f[e[i].to][j]);
                }
        }
    for(int i=0;i<m;i++)
    {
        f[u][i+1]=max(f[u][i],f[u][i+1]);
        g[u][i+1]=max(g[u][i],g[u][i+1]);
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        add(x+1,y+1);
        add(y+1,x+1);
    }
    dp(1,1);
    printf("%d\n",g[1][m]);
    return 0;
}

转载于:https://www.cnblogs.com/lokiii/p/8510764.html