BZOJ 2197 [Usaco2011 Mar]Tree Decoration

BZOJ_2197

    如果子树中有某个节点不符合要求，即便根再怎么符合要求都是没有任何意义的，因此要优先安排好子树中节点使其符合要求，再考虑根节点。对于任何一棵子树来讲，如果所有孩子选择的点的总和仍然不足根的C值的话，那么就还要在孩子中选出一些节点，而且如果要选那么必然应该选T最小的点。

#include<stdio.h>
#include<string.h>
#include<algorithm>
#define MAXD 100010
#define MAXM 100010
typedef long long LL;
int N, first[MAXD], e, next[MAXM], v[MAXM], c[MAXD], t[MAXD];
LL ANS, size[MAXD];
void add(int x, int y)
{v[e] = y;next[e] = first[x], first[x] = e ++;
}
void init()
{int p;memset(first, -1, sizeof(first[0]) * (N + 1)), e = 0;for(int i = 1; i <= N; i ++){scanf("%d%d%d", &p, &c[i], &t[i]);if(p != -1) add(p, i);}
}
int dfs(int cur)
{int min = t[cur];size[cur] = 0;for(int i = first[cur]; i != -1; i = next[i]){min = std::min(min, dfs(v[i]));size[cur] += size[v[i]];}if(size[cur] < c[cur])ANS += (c[cur] - size[cur]) * min, size[cur] = c[cur];return min;
}
void solve()
{ANS = 0;dfs(1);printf("%lld\n", ANS);
}
int main()
{while(scanf("%d", &N) == 1){init();solve();}return 0;
}