直接线段树合并就好了
之后在线段树上二分贪心选取金额较少的
如果是左偏树的话就开一个大根堆,根和子树顺次合并,合并之后堆内所有元素总和如果大于\(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;
}