51nod 1378 夹克老爷的愤怒[贪心][树形dp?]

//发现以前没写清楚,补一下坑
假如是链上的话就直接贪心搞,每隔$2*k$放一个守卫
现在搞到树上
递归处理,f[i]表示节点i的子树已经全受控制,还可以向上控制f[i]个单位
搜到当前位置需要处理的情况:
1.如果$f[i]\leq-k$,则答案加一
2.对所有子节点的f[i]取到min和max值,若$min+max>0$说明有一个子树中的节点放了可以镇压别的子树,当前f[i]=max-1
否则让长辈们解决(误),f[i]=min-1

#include<cstdio>
#include<vector>
#include<algorithm>
#define N 100000
using namespace std;

inline char nc(){
    static char buf[100000],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}

inline int read(){
    char c=nc(),b=1;int x;
    for(;!(c>='0'&&c<='9');c=nc()) if(c=='-') b=-1;
    for(x=0;c>='0'&&c<='9';x=x*10+c-'0',c=nc());
    return x*b;
}

vector<int>e[N+5];
int n,k,ans,f[N+5];

void dfs(int x,int fa){
    int minn=2e9,maxn=-2e9;
    for(int i=0;i<e[x].size();++i)
        if(e[x][i]!=fa){
            dfs(e[x][i],x);
            minn=min(minn,f[e[x][i]]);
            maxn=max(maxn,f[e[x][i]]);
        }
    if(minn==2e9) f[x]=-1;
    else if(minn<=-k) {
        ++ans;
        f[x]=k;
    }
    else if(minn+maxn>0) f[x]=maxn-1;
    else f[x]=minn-1;
}

int main(){
    n=read(),k=read();
    for(int i=1,x,y;i<n;++i){
        x=read()+1,y=read()+1;
        e[x].push_back(y),e[y].push_back(x);
    }
    if(k==0) {printf("%d\n",n);return 0;}
    dfs(1,0);
    if(f[1]<0) ++ans;
    printf("%d\n",ans);
}