[APIO2012]派遣

看着\(dp\)标签进来的。
结果是个数据结构嘛。
这个对于每个人都维护一下子树里\(sum <= m\)的最大人数就好了。
考虑贪心一下就不说了。从叶子节点往上合并堆,当然这种简单的合并操作,\(stl\)最好,跑得还很快。

[APIO2012]派遣
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<queue>
#define ll long long
#define N 100005

ll n,m;

struct P{
ll f,c,l;
ll head,sum;
P(){sum = 0;}
}T[N];

std::priority_queue<ll>QWQ[N];

ll head[N],cnt,ans;

struct E{int to,next;}e[N << 1];

inline void add(int x,int y){
	e[++cnt].to = y;
	e[cnt].next = head[x];
	head[x] = cnt;
}

inline void merge(int x,int y){//x -> y
	if(QWQ[T[x].head].size() > QWQ[T[y].head].size())
	std::swap(T[x].head,T[y].head),std::swap(T[x].sum,T[y].sum);
	while(QWQ[T[x].head].size())
	QWQ[T[y].head].push(QWQ[T[x].head].top()),T[y].sum += QWQ[T[x].head].top(),QWQ[T[x].head].pop();
}

inline void dfs(int u,int f){
//	std::cout<<u<<" "<<f<<std::endl;
	if(T[u].c <= m)
	QWQ[T[u].head].push(T[u].c),T[u].sum += T[u].c;
	for(int i = head[u];i;i = e[i].next){
		int v = e[i].to;
		if(v == f)
		continue;
		dfs(v,u);
		merge(v,u);
	}
	while(T[u].sum > m)
	T[u].sum -= QWQ[T[u].head].top(),QWQ[T[u].head].pop();
	ans = std::max((1ull) * ans, T[u].l * QWQ[T[u].head].size());
}

int main(){
	scanf("%lld%lld",&n,&m);
	for(int i = 1;i <= n;++i)
		scanf("%lld%lld%lld",&T[i].f,&T[i].c,&T[i].l),add(T[i].f,i);
	for(int i = 1;i <= n;++i)
	T[i].head = i;
	dfs(1,0);
	std::cout<<ans<<std::endl;
}
posted @ 2021-05-26 20:18  fhq_treap  阅读(34)  评论(0编辑  收藏  举报