zoj3279 简单二分

来源：互联网发布：哈利波特魔杖淘宝编辑：程序博客网时间：2024/05/18 01:24

随意在书上挑了道题，结果是道简单的二分题，随手写了。

题意：

有1到n 那个level 每一个level有a[i]只蚂蚁
两种操作

p a b 把第a个level的蚂蚁数量改成b

q a 查询第a只蚂蚁在哪个level里。

思路：要查的第x只蚂蚁，满足a[1]+a[2]+....+a[i-1]<x<a[1]+....a[i].所以1到n二分i就可以了，然后用线段树维护前缀和即可，当然也可以用树状数组，我习惯写线段树。

代码：

#include<iostream>#include<stdio.h>#include<string.h>#define lson l,mid,rt<<1#define rson mid+1,r,rt<<1|1using namespace std;const int maxn=100005;int sum[maxn<<2];void PushUP(int rt) {sum[rt] = sum[rt<<1] + sum[rt<<1|1];}void build(int l,int r,int rt) {if (l == r) {scanf("%d",&sum[rt]);return ;}int mid= (l + r) >> 1;build(lson);build(rson);PushUP(rt);}void update(int p,int add,int l,int r,int rt) {if (l == r) {sum[rt] = add;return ;}int mid= (l + r) >> 1;if (p <= mid) update(p , add , lson);else update(p , add , rson);PushUP(rt);}int query(int L,int R,int l,int r,int rt) {if (L <= l && r <= R) {return sum[rt];}int mid= (l + r) >> 1;int ret = 0;if (L <= mid) ret += query(L , R , lson);if (R > mid) ret += query(L , R , rson);return ret;}int main(){    int n,m;    char a[10];    int x,y;    while(~scanf("%d",&n)){        build(1,n,1);        scanf("%d",&m);        for(int i=1;i<=m;i++){            scanf("%s",a);            if(a[0]=='p'){                scanf("%d%d",&x,&y);                update(x,y,1,n,1);            }            else{                scanf("%d",&x);                int l=1,r=n,mid;                int ans;                while(l<r){                  mid=(l+r)>>1;                  ans=query(1,mid,1,n,1);                  if(ans<x) l=mid+1;                  if(ans>=x) r=mid;                }                printf("%d\n",r);            }        }    }return 0;}

0 0