标签:消息 wap ast 整数 namespace 数据 贪心 scan 节点
题目描述
输入
输出
样例输入
5 4
0 3 3
1 3 5
2 2 2
1 2 4
2 3 1
样例输出
6
题目大意
给你一棵以1为根的树,对于每个节点的子树,找出尽量多的点,使得这些点的点权之和不超过m,并把选出节点的个数与该节点的另一种权值的乘积更新到ans中,求ans的最大值。
题解
贪心+可并堆
首先有最简单的贪心法则:对于每个子树,选择点权尽量小的点。
由于题目中的m是个固定值,所以在贪心法则下,没有被选中的点对答案不会有任何贡献,可以看做直接删掉。
那么我们可以从下至上进行操作,每次把子节点子树中保留的点传到该节点上,然后再删点。
于是需要一种数据结构,支持删除最大值,以及将两个结构合并。
显然是可并堆。
每个点只被删1次,时间复杂度O(nlogn)。
#include <cstdio> #include <algorithm> using namespace std; int head[100010] , to[200010] , next[200010] , cnt , l[100010] , r[100010] , d[100010] , root[100010] , si[100010]; long long v[100010] , w[100010] , sum[100010] , m , ans; void add(int x , int y) { to[++cnt] = y; next[cnt] = head[x]; head[x] = cnt; } int merge(int x , int y) { if(!x) return y; if(!y) return x; if(v[x] < v[y]) swap(x , y); r[x] = merge(r[x] , y); if(d[l[x]] < d[r[x]]) swap(l[x] , r[x]); d[x] = d[r[x]] + 1; return x; } void dfs(int x) { int i; root[x] = x , sum[x] = v[x] , si[x] = 1; for(i = head[x] ; i ; i = next[i]) dfs(to[i]) , sum[x] += sum[to[i]] , si[x] += si[to[i]] , root[x] = merge(root[x] , root[to[i]]); while(sum[x] > m) sum[x] -= v[root[x]] , si[x] -- , root[x] = merge(l[root[x]] , r[root[x]]); ans = max(ans , w[x] * si[x]); } int main() { int n , i , x; scanf("%d%lld" , &n , &m); for(i = 1 ; i <= n ; i ++ ) scanf("%d%lld%lld" , &x , &v[i] , &w[i]) , add(x , i); d[0] = -1; dfs(1); printf("%lld\n" , ans); return 0; }
【bzoj2809】[Apio2012]dispatching 贪心+可并堆
标签:消息 wap ast 整数 namespace 数据 贪心 scan 节点
原文地址:http://www.cnblogs.com/GXZlegend/p/6532793.html