[BZOJ2060][Usaco2010 Nov]Visiting Cows 拜访奶牛（树形dp）

题目描述

传送门

题解

f[i][0]表示i子树中不选i的最大和，f[i][1]表示选i。

代码

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
const int max_n=5e4+5;
const int max_e=max_n*2;
int n,x,y,f[max_n][2];
int tot,point[max_n],next[max_e],v[max_e];
inline void add(int x,int y){++tot;next[tot]=point[x];point[x]=tot;v[tot]=y;}
inline void treedp(int x,int fa){
    bool flag=false;
    for (int i=point[x];i;i=next[i])
      if (v[i]!=fa){
        flag=true;
        treedp(v[i],x);
        f[x][1]+=f[v[i]][0];
        f[x][0]+=max(f[v[i]][1],f[v[i]][0]);
      }
}
int main(){
    scanf("%d",&n);
    for (int i=1;i<n;++i) scanf("%d%d",&x,&y),add(x,y),add(y,x);
    for (int i=1;i<=n;++i) f[i][1]=1;
    treedp(1,0);
    printf("%d\n",max(f[1][0],f[1][1]));
}