【BZOJ5319】军训列队（主席树）

【BZOJ5319】军训列队（主席树）

题面

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));
	}
}