P1552 [APIO2012]派遣

链接

https://www.luogu.org/problemnew/show/P1552

思路

忍者数量肯定越多越好
那就从下到上的合并它的孩子
左偏树的话
顺便维护一个tot，大头堆，如果tot大于了m，把大的删掉
如果左偏树忘干净了或者没学的话
线段树合并也是个不错的选择
直接权值线段树合并就好，内存30倍会炸，也许是我没离散化的缘故吧
查询在线段树上面二分

左偏树代码

#include <bits/stdc++.h>
#define FOR(i,a,b) for(int i=a;i<=b;++i)
#define ll long long
using namespace std;
const int maxn=100047;
inline int read() {int x=0,f=1;char s=getchar();for(;s>'9'||s<'0';s=getchar()) if(s=='-') f=-1;for(;s>='0'&&s<='9';s=getchar()) x=x*10+s-'0';return x*f;
}
struct edge {int v,nxt;
}e[maxn];
int head[maxn],tot;
void add_edge(int u,int v) {e[++tot].v=v;e[tot].nxt=head[u];head[u]=tot;
}
int n,m,mone;
ll sum[maxn],ans;
int size[maxn];
int ch[maxn][2],dis[maxn],val[maxn],xs[maxn];
int work(int a,int b) {if(!a||!b) return a+b;if(val[a]<val[b]) swap(a,b);ch[a][1]=work(ch[a][1],b);if(dis[ch[a][0]]<dis[ch[a][1]]) swap(ch[a][0],ch[a][1]);dis[a]=dis[ch[a][1]]+1;return a;
}
int merge(int x,int y) {int tmp=work(x,y);y=x^y^tmp,x=tmp;sum[x]+=sum[y];size[x]+=size[y];return tmp;
}
int delet(int x) {int tmp=work(ch[x][0],ch[x][1]);size[tmp]=size[x]-1;sum[tmp]=sum[x]-val[x];return tmp;
}
int dfs(int u) {int rt=u;for(int i=head[u];i;i=e[i].nxt) {int v=e[i].v;int tmp=dfs(v);rt=merge(rt,tmp);}while(sum[rt]>mone) rt=delet(rt);ans=max(ans,xs[u]*(ll)size[rt]);return rt;
}
int main() {n=read(),mone=read();int root;FOR(i,1,n) {int x=read();val[i]=read();xs[i]=read();sum[i]=val[i];size[i]=1;if(x) add_edge(x,i);else root=i;}int wuyong=dfs(root);cout<<ans<<"\n";return 0;
}

线段树合并代码

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <vector>
#include <utility>
#define ll long long
using namespace std;
const int N=1e5+7;
int read() {int x=0,f=1;char s=getchar();for(;s>'9'||s<'0';s=getchar()) if(s=='-') f=-1;for(;s>='0'&&s<='9';s=getchar()) x=x*10+s-'0';return x*f;
}
int n,m,money[N],leader[N],rt[N];
vector<int> G[N];
struct node {int l,r,siz;ll tot;
}e[N*30];
void pushup(int rt) {e[rt].siz=e[e[rt].l].siz+e[e[rt].r].siz;e[rt].tot=e[e[rt].l].tot+e[e[rt].r].tot;
}
int cnt;
void insert(int l,int r,int L,int &rt) {rt=++cnt;if(l==r) {e[rt].siz++;e[rt].tot+=l;return;}int mid=(l+r)>>1;if(L<=mid) insert(l,mid,L,e[rt].l);else insert(mid+1,r,L,e[rt].r);pushup(rt);
}
int merge(int l,int r,int x,int y) {if(!x||!y) return x+y;if(l==r) {e[x].siz+=e[y].siz;e[x].tot+=e[y].tot;return x;}int mid=(l+r)>>1;e[x].l=merge(l,mid,e[x].l,e[y].l);e[x].r=merge(mid+1,r,e[x].r,e[y].r);pushup(x);return x;
}
int query(int l,int r,int k,int rt) {if(l==r) return k>=e[rt].tot ? e[rt].siz : 0;int mid=(l+r)>>1;if(e[e[rt].l].tot>=k) return query(l,mid,k,e[rt].l);else return e[e[rt].l].siz+query(mid+1,r,k-e[e[rt].l].tot,e[rt].r);
}
ll ans;
void dfs(int u) {insert(1,1000000000,money[u],rt[u]);for(vector<int>::iterator it=G[u].begin();it!=G[u].end();++it) {dfs(*it);rt[u]=merge(1,1000000000,rt[u],rt[*it]);}ans=max(ans,(ll)leader[u]*query(1,1000000000,m,rt[u]));
}
int main() {n=read(),m=read();for(int i=1;i<=n;++i) {int x=read();money[i]=read();leader[i]=read();G[x].push_back(i);}dfs(1);printf("%lld",ans);return 0;
}