【[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;
}

标签：派遣,return,APIO2012,mid,int,query,now,define
来源： https://www.cnblogs.com/asuldb/p/10365010.html