树形dp模板 p2996

树形dp模板 p2996

洛谷原题

树形dp大部分在dfs上，可以不用结构体，而且主语nex和adj数组记得开二倍空间。

代码如下

#include <iostream>

using namespace std;
const int MAXN=60000;
int f[MAXN][2],nex[2*MAXN],adj[2*MAXN],head[2*MAXN],tot,n,a,b;
void addedge (int a, int b)
{
    nex[++tot] = head[a];
    head[a] = tot;
    adj[tot] = b;
}
void dfs(int nod,int fa)
{
    f[nod][1]=1;//******
    for(int i=head[nod];i;i=nex[i])//******
    {
        if(adj[i]==fa)
        {
            continue;
        }
        dfs(adj[i],nod);//******
        f[nod][0]=f[nod][0]+max(f[adj[i]][0],f[adj[i]][1]);
        f[nod][1]=f[nod][1]+f[adj[i]][0];
    }
}
int main()
{
    cin>>n;
    for(int i=1;i<n;i++)
    {
        int a,b;
        cin>>a>>b;
        addedge(a,b);
        addedge(b,a);
    }
    dfs(1, 0);//******
    cout<<max(f[1][0],f[1][1])<<endl;
    return 0;
}