Luogu P4559 [JSOI2018]列队 (区间问题, 主席树)

题目:

https://www.luogu.org/problemnew/show/P4559

一句话:

教官会指定一个区间 [l,r] 和集合点 K ，所有编号在 [l,r] 内的学生都必须赶到集合点列队。

简单来说,就是在主席树中我们要知道 这个区间的学生要向什么方向移动,知道了向什么方向移动就好办了,

还要知道这个区间的学生要移动到什么位置.

 

就是一个区间的学生移动到给定的区间,这个区间是连续的. Query 中有表示, 

[l  r] 中所有的学生全体移动到 [k,c] 这个些位置.剩下的计算,画一画就知道了.

 

#include<bits/stdc++.h>
using namespace std;
const int N = 1e6+100;
const int M =  N * 40;
int ls[M],rs[M],cnt,rt[N],num[M];
long long val[M],ans1,ans2,num1,num2;
int n,m,a[N];
void modify(int &now, int l, int r, int  k){
    val[++cnt] = val[now] + k;  num[cnt] = num[now] + 1; 
    ls[cnt] = ls[now]; rs[cnt] = rs[now]; now = cnt;
    if (l + 1 == r) return;
    int mid = (l + r) >> 1;
    if (k < mid) modify(ls[now],l,mid,k);
    if (k >= mid) modify(rs[now],mid,r,k);
}
long long Query(int rt1, int rt2, int l, int r, int k, int c){
    if (k > c) return 0;
    long long ans = 0;
    if (r-1 <= c){
        ans2 = val[rt2] - val[rt1]; 
        return 1ll*(c - k + 1) * c - ans2 - 1ll*(c - k)*(c-k+1)/2;
    }
    if (l >= k){
        ans1 = val[rt2] - val[rt1];
        return ans1 - 1ll*(c - k + 1) * k - 1ll*(c - k)*(c-k+1)/2;
    }
    int mid = (l + r) >> 1;
    int cc = num[ls[rt2]] - num[ls[rt1]];
    ans += Query(ls[rt1],ls[rt2],l,mid,k,k+cc-1);
    ans += Query(rs[rt1],rs[rt2],mid,r,k+cc,c);
    return ans;
}
int main(){
    int tot = 0,x,y,k;
    scanf("%d%d",&n,&m);
    for (int i = 1; i <= n; ++i) scanf("%d",&a[i]),tot = max(tot,a[i]);
    tot += 1e6 + 100;//之后要改的,为了方便修改而设成的值.
    for (int i = 1; i <= n; ++i){
        rt[i] = rt[i-1]; 
        modify(rt[i],1,tot+1,a[i]);
    }
    for (int i = 0; i < m; ++i){
        scanf("%d%d%d",&x,&y,&k);
        printf("%lld\n",Query(rt[x-1],rt[y],1,tot+1,k,k+y-x));
    }
    return 0;
}
/*
5 5
1 5 7 6 2
1 1 2
1 5 3
1 3 9
2 4 2
3 5 5

*/