HDU1899 Sum the K-th's(树状数组)

枚举，每次增加点，删除点

#include<cstdio>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<string>
#include<algorithm>
#include<map>
#include<queue>
#include<vector>
#include<cmath>
#include<utility>
using namespace std;
typedef long long LL;
const int N = , INF = 0x3F3F3F3F, MOD =  ;
#define MS(a, num) memset(a, num, sizeof(a))
int C[N];
int n, m, k;
int a[N], rak[N];

struct data{
    int num, i;
}b[N];
inline int lowbit(int x){
    return x&-x;
}
inline void add(int x, int val){//将第x个数增加val，从1计数
    for(int i=x;i<=n;i+=lowbit(i)){
        C[i] += val;
    }
}
inline int sum(int x){//求1到x的和
    int ret = ;
    for(int i=x;i>;i-=lowbit(i)){
        ret+=C[i];
    }
    return ret;
}

bool cmp(const data &a, const data &b){
    return a.num < b.num;
}

int getK(int val){
    int l = , r = n + ;
    while(l < r){
        int m = (l + r)>>;
        int s = sum(m);
        if(s < k){
            l = m + ;
        }else{
            r = m;
        }
    }
    return b[l -].num;
}
int main(){
    int t;
    cin>>t;
    while(t--){
        scanf("%d %d %d", &n, &m, &k);
        for(int i = ; i < n; i++){
            scanf("%d", &a[i]);
            b[i].num = a[i];
            b[i].i = i;
        }
        MS(C, );
        sort(b, b + n, cmp);

        for(int i = ; i< n; i++){
            rak[b[i].i] = i + ;
        }

        LL ans = ;
        for(int i = ; i < m; i++){
            add(rak[i % n], );
        }

        for(int i = m + ; i <  * m +  ;i++){
            add(rak[i % n], );
        }

        for(int i = m ; i <= m + n - ; i++){
            ans += getK(a[i % n]);
            ans %= MOD;
            if(i != m + n - ){
                add(rak[(i - m)%n], -);
                add(rak[(i + m + ) % n], );
                add(rak[i % n], );
                add(rak[(i + ) % n], -);
            }
        }
        printf("%I64d\n", ans % MOD);
    }
    return ;

}