操作格子(单点修改线段树)

问题描述

有n个格子，从左到右放成一排，编号为1-n。

共有m次操作，有3种操作类型：

1.修改一个格子的权值，

2.求连续一段格子权值和，

3.求连续一段格子的最大值。

对于每个2、3操作输出你所求出的结果。

输入格式

第一行2个整数n，m。

接下来一行n个整数表示n个格子的初始权值。

接下来m行，每行3个整数p,x,y，p表示操作类型，p=1时表示修改格子x的权值为y，p=2时表示求区间[x,y]内格子权值和，p=3时表示求区间[x,y]内格子最大的权值。

输出格式

有若干行，行数等于p=2或3的操作总数。

每行1个整数，对应了每个p=2或3操作的结果。

样例输入

4 3
1 2 3 4
2 1 3
1 4 3
3 1 4

样例输出

6
3

数据规模与约定

对于20%的数据n <= 100，m <= 200。

对于50%的数据n <= 5000，m <= 5000。

对于100%的数据1 <= n <= 100000，m <= 100000，0 <= 格子权值 <= 10000。

写线段树可千万要注意边界问题，调吐了。。。

#include <stdio.h>
#define MAX 150005
#define MOD 1000000007
typedef long long LL;

//const int INF = 1e9+7;
#define VM 1000010
int a[VM];
int st[VM<<2],mt[VM<<2];
int max(int a,int b){return a>b?a:b;}
void pushup(int rt){
    st[rt]=st[rt<<1]+st[rt<<1|1];
    mt[rt]=max(mt[rt<<1],mt[rt<<1|1]);
}
void build(int rt,int l,int r){
    if(l==r){
        st[rt]=a[l];
        mt[rt]=a[l];
        return ;
    }
    else{
        int m=(l+r)>>1;
        build(rt<<1,l,m);
        build(rt<<1|1,m+1,r);
        pushup(rt);
        //pushmax(rt);
    }
    return ;
}
void update(int rt,int l,int r,int ll,int ans){
    if(l==r&&l==ll){
        st[rt]=ans;
        mt[rt]=ans;
        return ;
    }
    int m=(l+r)>>1;
    if(ll<=m)update(rt<<1,l,m,ll,ans);
    if(ll>m)update(rt<<1|1,m+1,r,ll,ans);
    pushup(rt);
    //pushmax(rt);
}
int query(int rt,int l,int r,int ll,int rr){
    if(ll==l&&rr==r)return mt[rt];
    int m=(l+r)>>1;
    if(rr<=m)return query(rt<<1,l,m,ll,rr);
    else if(ll>m)return query(rt<<1|1,m+1,r,ll,rr);
    else {
        return max(query(rt<<1,l,m,ll,m),query(rt<<1|1,m+1,r,m+1,rr));
    }
}
int querysum(int rt,int l,int r,int ll,int rr){
    if(ll==l&&rr==r)return st[rt];
    int m=(l+r)>>1;
    if(rr<=m)return querysum(rt<<1,l,m,ll,rr);
    else if(ll>=m+1)return querysum(rt<<1|1,m+1,r,ll,rr);
    else {
        return querysum(rt<<1,l,m,ll,m)+querysum(rt<<1|1,m+1,r,m+1,rr);
    }
}
int main()
{
    int n,m,i,j;
    int x,y,p;
    scanf("%d%d",&n,&m);
    for(i=1;i<=n;i++)scanf("%d",&a[i]);
    build(1,1,n);
    while(m--){
        scanf("%d%d%d",&p,&x,&y);
        if(p==1){
            update(1,1,n,x,y);
        }
        else if(p==2){
            printf("%d\n",querysum(1,1,n,x,y));
        }
        else if(p==3){
            printf("%d\n",query(1,1,n,x,y));
        }
    }
    return 0;
}