51nod 1378 夾克老爺的憤怒[貪心][樹形dp?]

//發現以前沒寫清楚,補一下坑
假如是鏈上的話就直接貪心搞,每隔2∗k 放一個守衛
現在搞到樹上
遞歸處理,f[i]表示節點i的子樹已經全受控制,還可以向上控制f[i]個單位
搜到當前位置需要處理的情況:
1.如果f[i]≤−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);
}