传送门: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