bzoj5072 小A的树 - 树上背包

我们发现对于一个子树，若其最少包含的黑色点数为a1，最多包含的黑色点数为a2，则在这棵子树中，能够包含的黑色点数可以是a1~a2之间的任何数，因为每加入一个点，黑色点数量最多加1个

我们需要求出对于每个询问结点量x，其最多和最少包含的黑色结点数，如果询问的黑色节点数属于这个区间，则满足

需要在树上做背包，设f[i][j]为以i为根的子树中选出j个点所能包含的最多黑色结点数，g[i][j]为以i为根的子树中选出j个点所能包含的最少黑色结点数

转移则是在某一棵子树遍历完后，先不将这棵子树的所有结点算到根上，而是分为两部分：1.这棵子树，2.根和其他子树（“其他子树”已经在“这棵子树”之前被遍历），这样我们可以枚举1部分和2部分各自选了j个点和k个点，然后去更新大小为j+k的背包

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;
#define debug(x) cerr << #x << "=" << x << endl;
const int MAXN = 5010;
const int MAXM = 5010*2 - 2;
int n,q,vis[MAXN],son[MAXN],f[MAXN][MAXN],b[MAXN],last[MAXN],tot,t,g[MAXN][MAXN];
int ansf[MAXN], ansg[MAXN];
bool flg;
struct Edge{
    int u,v,to;
    Edge(){}
    Edge(int u, int v, int to) : u(u), v(v), to(to) {}
}e[MAXM];

inline void add(int u, int v) {
    e[++tot] = Edge(u,v,last[u]);
    last[u] = tot;
}

inline void read(int &x) {
    x = 0;
    char ch = getchar();
    while(ch<'0'||ch>'9') ch = getchar();
    while(ch>='0'&&ch<='9') x = x*10+ch-'0', ch = getchar();
    return;
}
void dfs(int x) {
    vis[x] = 1;
    son[x] = 1;
    f[x][1] = g[x][1] = b[x];
    for(int i=last[x]; i; i=e[i].to) {
        int v = e[i].v;
        if(vis[v]) continue;
        dfs(v);
        for(int j=son[x]; j; j--) {//逆序枚举，每个点只可选一次
            for(int k=0; k<=son[v]; k++) {
                f[x][j+k] = max(f[x][j+k], f[x][j] + f[v][k]);
                g[x][j+k] = min(g[x][j+k], g[x][j] + g[v][k]);
            }
        }
        son[x] += son[v];//这句放到后面，因为我们把x的一棵子树和x的其他子树分离了
    }
    for(int i=1; i<=son[x]; i++) {
        ansf[i] = max(ansf[i], f[x][i]);
        ansg[i] = min(ansg[i], g[x][i]);
    }
}
void query(int x ,int y) {
    if(y>=ansg[x] && y<=ansf[x]) {
        printf("YES\n");
    } else {
        printf("NO\n");
    }
}
int main() {
    read(t);
    while(t--) {
        tot = 0;
        memset(last,0,sizeof(last));
        memset(vis,0,sizeof(vis));
        memset(son,0,sizeof(son));
        read(n);
        read(q);
        for(int i=1; i<=n-1; i++) {
            int u,v;
            read(u),read(v);
            add(u,v);
            add(v,u);
        }
        for(int i=1; i<=n; i++) {
            read(b[i]);
        }
        memset(f,0,sizeof(f));
        memset(g,0x3f,sizeof(g));
        memset(ansf,0,sizeof(ansf));
        memset(ansg,0x3f,sizeof(ansg));
        dfs(1);
        for(int i=1; i<=q; i++) {
            int x, y;
            read(x), read(y);
            query(x,y);
        }
        printf("\n");
    }
    return 0;
}