【[APIO2012]派遣】

题目

直接线段树合并就好了

之后在线段树上二分贪心选取金额较少的

如果是左偏树的话就开一个大根堆，根和子树顺次合并，合并之后堆内所有元素总和如果大于\(m\)就删除堆顶，由于每个元素只会被删除一次，所以复杂度非常科学

代码

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#define maxn 100005
#define M 4000005
#define re register
#define LL long long
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
inline LL read()
{
	char c=getchar();LL x=0;while(c<'0'||c>'9') c=getchar();
	while(c>='0'&&c<='9') x=(x<<3)+(x<<1)+c-48,c=getchar();return x;
}
struct E{int v,nxt;}e[maxn];
int n,head[maxn],cnt,num,R,rt[maxn],t[maxn],fa[maxn];
int l[M],r[M],d[M],sz;
LL s[M],m,a[maxn],b[maxn],c[maxn],ans;
inline void add(int x,int y) {e[++num].v=y;e[num].nxt=head[x];head[x]=num;}
inline int find(LL x) {
	int ll=1,rr=sz;
	while(ll<=rr) {
		int mid=ll+rr>>1;
		if(c[mid]==x) return mid;
		if(c[mid]>x) rr=mid-1;else ll=mid+1;
	}
	return 0;
}
int change(int now,int x,int y,int pos,LL val) {
	if(!now) now=++cnt;
	d[now]++,s[now]+=val;
	if(x==y) return now;
	int mid=x+y>>1;
	if(pos<=mid) l[now]=change(l[now],x,mid,pos,val);
		else r[now]=change(r[now],mid+1,y,pos,val);
	return now;
}
int merge(int a,int b,int x,int y) {
	if(!a) return b;if(!b) return a;
	if(x==y) {
		d[a]+=d[b],s[a]+=s[b];
		return a;
	}
	int mid=x+y>>1;
	l[a]=merge(l[a],l[b],x,mid),r[a]=merge(r[a],r[b],mid+1,y);
	d[a]=d[l[a]]+d[r[a]],s[a]=s[l[a]]+s[r[a]];
	return a;
}
int query(int now,int x,int y,LL k) {
	if(x==y) {
		if(!d[now]) return 0;
		return min(d[now],k/c[x]);
	}
	int mid=x+y>>1;
	if(k>s[l[now]]) return query(r[now],mid+1,y,k-s[l[now]])+d[l[now]];
	return query(l[now],x,mid,k);
}
void dfs(int x) {
	for(re int i=head[x];i;i=e[i].nxt) {
		dfs(e[i].v);
		rt[x]=merge(rt[x],rt[e[i].v],1,sz);
	}
	int now=query(rt[x],1,sz,m);
	ans=max(ans,(LL)now*b[x]);
}
int main()
{
	n=read(),m=read();
	for(re int i=1;i<=n;i++) {
		fa[i]=read(),c[i]=a[i]=read(),b[i]=read();
		if(!fa[i]) R=i;else add(fa[i],i);
	}
	std::sort(c+1,c+n+1);
	sz=std::unique(c+1,c+n+1)-c-1;
	for(re int i=1;i<=n;i++) t[i]=find(a[i]);
	for(re int i=1;i<=n;i++) rt[i]=change(rt[i],1,sz,t[i],a[i]);
	dfs(R);
	query(rt[1],1,sz,m);
	printf("%lld\n",ans);
	return 0;
}