hdu4217 Data Structure?

题目链接：点击打开链接

树状数组，每个区间保存此区间中有多少未取过的数，每次输入k，查区间内刚好是k个数的点，然后update（k，-1）。

对区间更新对点查询，对区间（a，b）加v操作：update（a，v），update（b，-v），对x点查询：正常quary（x）。

本题每次都更新至区间结尾，所以不需要update（b，-v）。

当需要确定第一个达到某值的点时，可使用：

int quary(int n){
    int cur=0;
    int k=0;
    for(int i=18;i>=0;i--){
           if(k+(1<<i)<=N&&cur+c[k+(1<<i)]<n){
                k+=(1<<i);
                cur+=c[k];
           }
    }
    return k+1;
}

cur+c[k+(1<<i)]<n)此处必须为小于而不是小于等于，因为得到的是应是第一个达到某值的，而算法是从大到小判定的，结果位置应是最接近某值的数+1。

结果用long long。。

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#define FOR(i, x, y) for(int i = x; i <= y; i++)
#define MS(x, y) memset(x, y, sizeof(x))
using namespace std;
int c[277777];
int N,T,K;
int lowbit(int n){
    return (n&(-n));
}
void update(int n,int v){
    while(n<=N){
        c[n]+=v;
        n+=lowbit(n);

    }
}

int quary(int n){
    int cur=0;
    int k=0;
    for(int i=18;i>=0;i--){
           if(k+(1<<i)<=N&&cur+c[k+(1<<i)]<n){
                k+=(1<<i);
                cur+=c[k];
           }
    }
    return k+1;
}

int main(){
    int T;
    cin>>T;
    int cas=0;
    while(T--){
            scanf("%d%d",&N,&K);
            FOR(i,0,N){
                c[i]=0;
            }
            FOR(i,1,N){
                update(i,1);
            }
            int t;
            long long res=0;
            FOR(i,1,K){
                scanf("%d",&t);
                int k=quary(t);
                res+=k;
                update(k,-1);
            }
            cout <<"Case "<<++cas<<": "<<res<<endl;
    }
    return 0;
}