H Cutting Bamboos 2019牛客多校第9场

传送门:https://ac.nowcoder.com/acm/contest/889/H

这题主席树板题竟然只过了这么点人。。。

我们算出从1-x次cut,需要切掉的竹子总长度是多少,然后二分答案mid,看大于mid 的竹子有多少,已经从mid切下去能切出总共多少长度的竹子。

以权值为下标,每个点存储竹子个数和长度总和的主席树就每个跟节点rt[i]代表 1--i的竹子的情况。

然后每次二分出一个mid,那么只有>mid的竹子会被切到,那么我们就去找 [low, 1e5]的竹子去减这个mid得到总和。

#include
#define maxl 200010
#define eps 1e-8
using namespace std;
 
int n,q,tot,up=1e5;
int h[maxl],rt[maxl*41];
long long sum[maxl];
double mid;
struct node
{
    int ls,rs;
    int cnt;
    long long sum;
}tree[maxl*41];
 
inline void insert(int num,int &x,int l,int r)
{
    tree[++tot]=tree[x];x=tot;
    ++tree[x].cnt;tree[x].sum+=num;
    if(l==r) return;
    int mid=(l+r)>>1;
    if(num<=mid)
        insert(num,tree[x].ls,l,mid);
    else
        insert(num,tree[x].rs,mid+1,r);
}
 
inline void prework()
{
    for(register int i=1;i<=n;++i)
    {
        scanf("%d",&h[i]);
        sum[i]=sum[i-1]+h[i];
    }
    tot=0;tree[0].cnt=tree[0].sum=tree[0].ls=tree[0].rs=0;
    for(register int i=1;i<=n;++i)
    {
        rt[i]=rt[i-1];
        insert(h[i],rt[i],1,up);
    }
}
 
inline double query(int i,int j,int l,int r,int l1,int r1)
{
    double ret=0;
    if(l==l1 && r==r1)
    {
        ret=(tree[j].sum-tree[i].sum)-(tree[j].cnt-tree[i].cnt)*mid;
        return ret;
    }
    int mid=(l1+r1)>>1;
    if(r<=mid)
        ret=query(tree[i].ls,tree[j].ls,l,r,l1,mid);
    else if(l>mid)
        ret=query(tree[i].rs,tree[j].rs,l,r,mid+1,r1);
    else
    {
        ret=query(tree[i].ls,tree[j].ls,l,mid,l1,mid);
        ret+=query(tree[i].rs,tree[j].rs,mid+1,r,mid+1,r1);
    }
    return ret;
}
 
inline void mainwork()
{
    int ll,rr,x,y,low;
    double ss,l,r,ret;
    for(int i=1;i<=q;i++)
    {
        scanf("%d%d%d%d",&ll,&rr,&x,&y);
        ss=1.0*(sum[rr]-sum[ll-1])*x/y;
        l=0,r=up;
        while(l+eps1.0*low)
                low++;
            ret=query(rt[ll-1],rt[rr],low,up,1,up);
            if(ret

 

你可能感兴趣的:(主席树)