https://www.lydsy.com/JudgeOnline/problem.php?id=5319
https://www.luogu.org/problemnew/show/P4559
题面见上。
40pts：
不难想到询问应当在主席树上做。
[l,r]为我们询问的这些人所在位置的坐标区间，[l1,r1]为这些人最终要去这个区间里站着。
我们维护主席树节点代表的区间的人数，则设当前节点左子树人数为ls，则可以递归处理[l,mid][l1,l1+ls-1]和[mid+1,r][l1+ls,r1]，复杂度O(n^2)。
100pts：
然后我们就试图剪枝呗看看我们能剪成啥样。
我们能处理r<=l1和r1<=l的情况吗？当然可以。
我们举前一种情况为例（后面的情况基本同理）：
只需要先让这些人都跑到l1，然后再让这些人在[l1,r1]站好就行了。
于是算出这些人到l1的距离，再算这些人在[l1,r1]的体力值就行了，主席树维护当前人坐标之和即可做到前者，而后者就是等差数列求和。
其实写到这里就可以AC了（数据有点弱（笑））。
真·100pts：
那我们继续想，我们能处理r<=r1和l1<=l的情况吗？当然可以。
还是举前一种情况为例（后面的情况基本同理）：
只需要先让这些人都跑到r1，然后再让这些人在[l1,r1]站好就行了。
注意唯一不同的是"站好"这个过程实际上是“退流”——退体力值的过程，因为我们能够证明这种情况下每个人只可能跑多不可能跑少，所以我们前者减后者即可，维护方法同上。
那么复杂度到底是什么呢？思考一下不难发现，当mid<=l1+ls-1时候我们前子问题可以O(1)算出，后子问题O(log)继续递归，当mid>l1+ls-1时后子问题可以O(1)算出，前子问题O(log)继续递归。
复杂度O(nlog)可以通过本题。
#include<cmath> #include<queue> #include<vector> #include<cstdio> #include<cctype> #include<cstring> #include<iostream> #include<algorithm> using namespace std; typedef long long ll; const int N=5e5+5; const int M=1e6; inline int read(){ int X=0,w=0;char ch=0; while(!isdigit(ch)){w|=ch=='-';ch=getchar();} while(isdigit(ch))X=(X<<3)+(X<<1)+(ch^48),ch=getchar(); return w?-X:X; } struct tree{ int l,r,sum; ll dis; }tr[M*20+5]; int n,m,pool,rt[N]; void insert(int y,int &x,int l,int r,ll a){ tr[x=++pool]=tr[y]; tr[x].sum++;tr[x].dis+=a; if(l==r)return; int mid=(l+r)>>1; if(a<=mid)insert(tr[y].l,tr[x].l,l,mid,a); else insert(tr[y].r,tr[x].r,mid+1,r,a); } ll query(int nl,int nr,int l,int r,int l1,int r1){ if(l1>r1)return 0; if(r<=r1){ return (ll)r1*(r1-l1+1)-(tr[nr].dis-tr[nl].dis)-(ll)(r1-l1)*(r1-l1+1)/2; } if(l1<=l){ return (tr[nr].dis-tr[nl].dis)-(ll)l1*(r1-l1+1)-(ll)(r1-l1)*(r1-l1+1)/2; } int mid=(l+r)>>1; int delta=tr[tr[nr].l].sum-tr[tr[nl].l].sum; return query(tr[nl].l,tr[nr].l,l,mid,l1,l1+delta-1) +query(tr[nl].r,tr[nr].r,mid+1,r,l1+delta,r1); } int main(){ n=read(),m=read(); for(int i=1;i<=n;i++)insert(rt[i-1],rt[i],1,M,read()); for(int i=1;i<=m;i++){ int l=read(),r=read(),k=read(); printf("%lld ",query(rt[l-1],rt[r],1,M,k,k+r-l)); } return 0; }
+++++++++++++++++++++++++++++++++++++++++++
+本文作者：luyouqi233。 +
+欢迎访问我的博客：http://www.cnblogs.com/luyouqi233/+
+++++++++++++++++++++++++++++++++++++++++++