【LOJ】#2118. 「HEOI2015」兔子与樱花

题解

怎么觉得都像树dp，不像贪心

但是树dp确实做不了

把每个节点的值设置为樱花+儿子数

把儿子合并到父亲上就是父亲的剩余容量加上儿子的值-1

每次在父亲的时候将儿子的值排序然后能加就加上

因为儿子如果不加进去那么之后的操作与儿子再也没有关系了，儿子影响的只有父亲，那么只是能让父亲一个节点被加入，那么和现在就加入儿子是等价的

代码

#include <bits/stdc++.h>
#define fi first
#define se second
#define pii pair<int,int>
#define mp make_pair
#define pb push_back
#define enter putchar('\n')
#define space putchar(' ')
#define MAXN 2000005
//#define ivorysi
using namespace std;
typedef long long int64;
typedef long double db;
typedef unsigned int u32;
template<class T>
void read(T &res) {
    res = 0;char c = getchar();T f = 1;
    while(c < '0' || c > '9') {
	if(c == '-') f = -1;
	c = getchar();
    }
    while(c >= '0' && c <= '9') {
	res = res * 10 + c - '0';
	c = getchar();
    }
    res *= f;
}
template<class T>
void out(T x) {
    if(x < 0) {putchar('-');x = -x;}
    if(x >= 10) out(x / 10);
    putchar('0' + x % 10);
}
int N,M,val[MAXN],ans,id[MAXN],cnt;
struct node {
    int to,next;
}E[MAXN * 2];
int head[MAXN],sumE;
void add(int u,int v) {
    E[++sumE].to = v;
    E[sumE].next = head[u];
    head[u] = sumE;
}
bool cmp(int a,int b) {
    return val[a] < val[b];
}
void Init() {
    read(N);read(M);
    for(int i = 1 ; i <= N ; ++i) {
	read(val[i]);
    }
    int k,p;
    for(int i = 1 ; i <= N ; ++i) {
	read(k);val[i] += k;
	for(int j = 1 ; j <= k ; ++j) {
	    read(p);++p;add(i,p);
	}
    }
}
void dfs(int u) {
    for(int i = head[u] ; i ; i = E[i].next) {
	int v = E[i].to;
	dfs(v);
    }
    cnt = 0;
    for(int i = head[u] ; i ; i = E[i].next) {
	int v = E[i].to;id[++cnt] = v;
    }
    sort(id + 1,id + cnt + 1,cmp);
    for(int i = 1 ; i <= cnt ; ++i) {
	if(M - val[u] >= val[id[i]] - 1) {val[u] += val[id[i]] - 1;++ans;}
    }
}
int main() {
#ifdef ivorysi
    freopen("f1.in","r",stdin);
#endif
    Init();
    dfs(1);
    out(ans);enter;
}