ZOJ 3279 Ants 二分树状数组

----------

const int maxn=110000;
const int maxm=10000;
int n,m;
struct BIT{
    int n;
    int tree[maxn];
    void init(int n){
        this->n=n;
        memset(tree,0,sizeof(tree));
    }
    int lowbit(int x){
        return x&(-x);
    }
    void add(int x,int val){
        for (int i=x;i<=n;i+=lowbit(i)) tree[i]+=val;
    }
    int query(int x){
        int ret=0;
        for (int i=x;i>0;i-=lowbit(i)) ret+=tree[i];
        return ret;
    }
    //离散 p=lower_bound(b+1,b+n+1,a[i])-b;
    //逆序数 x=(i-1)-query(p);add(p,1);
}bt;
int a[maxn];
int main(){
    while (~scanf("%d",&n)){
        bt.init(n);
        for (int i=1;i<=n;i++) {
            scanf("%d",&a[i]);
            bt.add(i,a[i]);
        }
        scanf("%d",&m);
        char s[2];
        while (m--){
            scanf("%s",s);
            if (s[0]=='p'){
                int x,y;
                scanf("%d%d",&x,&y);
                bt.add(x,-a[x]);
                bt.add(x,y);
                a[x]=y;
            }
            else if (s[0]=='q'){
                int x,l=1,r=n,ans=0;
                scanf("%d",&x);
                while (l<=r){
                    int mid=(l+r)/2;
                    if (bt.query(mid)<x){
                        ans=mid;
                        l=mid+1;
                    }
                    else r=mid-1;
                }
                printf("%d\n",ans+1);
            }
        }
    }
	return 0;
}

----------