hdu 1754 I Hate It (树状数组版)

题目链接:点击打开链接


解释一下两个函数:

add:先将原数组中a[p]改为n,再把c数组中所有包括a[p]的元素都与n比较大小,每次p加lowbit(n)即可找到下一个包括a[p]的元素。

query:找l,r区间内最大值:

              step 1:首先令初值为a[r],然后令r-=1,因为r-1是最近的一个可能是此区间最值的位置,然后判断此时r-l是否大于等于lowbit(r)。

              step  2:如果是则说明c[r]包含的整个区间都在l,r之间,则直接比较当前最大值和此区间最大值,然后令r-=lowbit(r)找到下个区间,继续判断这个区间是否整个在l,r之间。

             step 3: 如果不是,则说明这个区间并不全部在l,r之间,不能直接与c[r]比较,则先与a[r]比较(a[r]一定在l,r之间),然后回到step 1,令r-=1,直到l==r对整个区间判断完毕。


发现一个问题是用gets读字符串会读入之前留下的回车,而用scanf("%s")读入则不会。

代码:

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
int c[200010];
int a[200010];
int lowbit(int n){
    return n&(-n);
}
int N,M;
void add(int p,int n){
    a[p]=n;
    while(p<=N){
            c[p]=max(c[p],n);
            p+=lowbit(p);
    }
}

int query(int l,int r){
    int res=a[r];
    while(l!=r){
        for(r-=1;r-l>=lowbit(r);r-=lowbit(r)){
            res=max(res,c[r]);
        }
        res=max(res,a[r]);
    }
    return res;
}

int main(){
    while(~scanf("%d%d",&N,&M)){
        memset(c,0,sizeof(c));
        for(int i=1;i<=N;i++){
            int tmp;
            scanf("%d",&tmp);
            add(i,tmp);
        }

        char tmp[10];
        int s,e;
        for(int i=1;i<=M;i++){
            scanf("%s%d%d",tmp,&s,&e);
            if(tmp[0]=='U'){
                add(s,e);
            }
            else
                printf("%d\n",query(s,e));
        }
    }
    return 0;
}




你可能感兴趣的:(树状数组)