【BZOJ5319】军训列队(主席树)

232 篇文章 0 订阅
19 篇文章 0 订阅

题面

BZOJ
洛谷

题解

一眼题既视感。。。
首先很明显,每次询问的结果显然是做一次离散。
然后直接上主席树就好了。。。
查询答案的方式也很简单
考虑一下那个绝对值是个什么东西
如果所有的点都在目标区间以左
则直接区间和去算一下。
如果所有的点都在目标区间以右
好像和在左边一样的。
否则,把区间隔开计算就好了。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
#define Ln 1000000
#define MAX 555555
#define ll long long
inline int read()
{
    int x=0,t=1;char ch=getchar();
    while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
    if(ch=='-')t=-1,ch=getchar();
    while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
    return x*t;
}
struct Node{int v,ls,rs;ll s;}t[MAX<<5];
int rt[MAX],n,m,S[MAX],a[MAX],len,tot;
void Modify(int &x,int ff,int l,int r,int p)
{
    t[x=++tot]=t[ff];t[x].v++;t[x].s+=p;
    if(l==r)return;int mid=(l+r)>>1;
    if(p<=mid)Modify(t[x].ls,t[ff].ls,l,mid,p);
    else Modify(t[x].rs,t[ff].rs,mid+1,r,p);
}
ll Query(int A,int B,int l,int r,int L,int R)
{
    if(L>R)return 0;
    if(r<=L)return 1ll*(L+R)*(R-L+1)/2-(t[B].s-t[A].s);
    if(l>=R)return (t[B].s-t[A].s)-1ll*(L+R)*(R-L+1)/2;
    int mid=(l+r)>>1,peo=t[t[B].ls].v-t[t[A].ls].v;
    return Query(t[A].ls,t[B].ls,l,mid,L,L+peo-1)+Query(t[A].rs,t[B].rs,mid+1,r,L+peo,R);
}
int main()
{
    n=read();m=read();
    for(int i=1;i<=n;++i)Modify(rt[i],rt[i-1],1,Ln,read());
    while(m--)
    {
        int l=read(),r=read(),K=read();
        printf("%lld\n",Query(rt[l-1],rt[r],1,Ln,K,K+r-l));
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值