BZOJ 1117 救火站Gas （贪心）

1117: [POI2009]救火站Gas

Time Limit: 10 Sec Memory Limit: 162 MB
Description

给你一棵树，现在要建立一些消防站，有以下要求: 1. 消防站要建立在节点上，每个节点可能建立不只一个消防站。 2. 每个节点应该被一个消防站管理，这个消防站不一定建立在该节点上。 3. 每个消防站可以管理至多s个节点。 4. 消防站只能管理距离（两点间最短路径的边数）不超过k的结点。请问至少要设立多少个消防站。

Input

第一行n,s,k。接下来n-1行每行xi,yi描述一条边连接xi与yi。 1<=n<=100000 1<=s<=n 1<=k<=20 1<=xi

Output

一个数，最少的消防站个数。

Sample Input

12 3 1
1 12
3 8
7 8
8 9
2 12
10 12
9 12
4 8
5 8
8 11
6 8

Sample Output

4

思路：
炸天的贪心，想不出来T.T。。。
不知道怎么才能讲清楚，大家感性认识吧。【摊手
贪心策略：以1为根建树，从叶子开始，能不放就不放，直到必须放了(有没被覆盖节点与当前节点距离为K)的时候再放。
在每个节点维护两个大小为K的数组。
第一个数组req记录这个节点的子树中还没被覆盖的点中距离为i的有多少个，第二个数组has记录这个节点的子树中控制范围为i的消防站还能再控制几个节点。
何时让这两个数组相互抵消？
当我们用一个剩余覆盖范围很广的消防站去覆盖一个离的很近的消防站显然是浪费，所以抵消的原则是：再过一步就抵消不了了的时候，才去抵消。也就是说，若这个站控制范围为i，我们只去抵消距离为i和i-1的节点。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <vector>
#include <algorithm>
#define LL long long
#define N 100010
using namespace std;

LL req[N][25], has[N][25];

int ans, idc=0;
int n, s, k;
int head[N];

struct Edge{
    int to, nxt;
}ed[N << 1];

void adde(int u, int v){
    ed[++idc].to = v;
    ed[idc].nxt = head[u];
    head[u] = idc;
}

void solve(int u){
    for(int i=0; i<=k; i++) {
        for(int j=i; j>=0 && j>=i-1; j--){//到fa时req++，has--，相差会改变2 
            LL cut = min(has[u][i], req[u][j]);
            has[u][i] -= cut; req[u][j] -= cut;
            if(!has[u][i]) break;//之前的操作保证了可以完全覆盖 
        }
    }
}

void dfs(int u, int fa){
    for(int i=head[u]; i; i=ed[i].nxt) {
        int v = ed[i].to;
        if(v == fa) continue;//req[u][j]表示u为root的子树中还没被覆盖的点中距离u为j的有多少个
        dfs(v, u);           //has[u][j]表示u为root的子树中可覆盖所有距离u为j的点的救火站还能覆盖几个节点
        for(int j=0; j<k; j++) req[u][j+1] += req[v][j];//离子树内的点远了一个单位 
        for(int j=1; j<=k; j++) has[u][j-1] += has[v][j];//离子树内救火站远了一个单位，子树内在临界情况的点被处理 
    }
    req[u][0]++;//以u为中心dis==0的范围内还需要覆盖req个点
    LL cc = req[u][k] - has[u][k];
    if(cc > 0){
        int num = (cc + s - 1) / s;//至少建立num个救火站 
        has[u][k] += num * s;//以u为中心dis<=k的范围内还可以覆盖has个点 
        ans += num;
    }
    solve( u );
}

int main() {
    scanf("%d%d%d", &n, &s, &k);
    for(int i=1; i<n; i++){
        int u, v; scanf("%d%d", &u, &v);
        adde(u, v); adde(v, u);
    }
    dfs(1, 0);
    for(int i=0; i<=k; i++) {
        for(int j=i; j>=0; j--){
            LL cut = min(has[1][i], req[1][j]);
            has[1][i] -= cut; req[1][j] -= cut;
            if(!has[1][i]) break;
        }
    }
    LL cc = 0;
    for(int i=0; i<=k; i++) cc += req[1][i];
    ans += (cc + s - 1) / s;
    printf("%d\n", ans);
    return 0;
}