操作格子 线段树

  算法训练 操作格子  

时间限制：1.0s   内存限制：256.0MB

      

问题描述

有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 <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <queue>
#include <stack>
#include <map>
#include <cmath>
#include <vector>
#define max_ 100010
#define inf 0x3f3f3f3f
#define ll long long
using namespace std;
struct node
{
    int l,r,w;
    int sum,maxx;
};
struct node tree[max_*4];
int n,m;
void built(int i,int l,int r)
{
    tree[i].l=l;
    tree[i].r=r;
    if(l==r)
    {
        cin>>tree[i].sum;
        tree[i].maxx=tree[i].sum;
        return;
    }
    int mid=(l+r)>>1;
    built(i<<1,l,mid);
    built(i<<1|1,mid+1,r);
    tree[i].sum=tree[i<<1].sum+tree[i<<1|1].sum;
    tree[i].maxx=max(tree[i<<1].maxx,tree[i<<1|1].maxx);
}
void updata(int i,int d,int w)
{
    if(tree[i].l==tree[i].r)
    {
        tree[i].sum=tree[i].maxx=w;
        return;
    }
    int mid=(tree[i].l+tree[i].r)>>1;
    if(d<=mid)
    updata(i<<1,d,w);
    else
    updata(i<<1|1,d,w);
    tree[i].sum=tree[i<<1].sum+tree[i<<1|1].sum;
    tree[i].maxx=max(tree[i<<1].maxx,tree[i<<1|1].maxx);
}
int query(int i,int l,int r,int t)
{
    if(tree[i].l==l&&tree[i].r==r)
    {
        if(t==2)
        return tree[i].sum;
        else if(t==3)
        return tree[i].maxx;
    }
    int mid=(tree[i].l+tree[i].r)>>1;
    if(r<=mid)
    return query(i<<1,l,r,t);
    else if(l>mid)
    return query(i<<1|1,l,r,t);
    else
    {
        if(t==2)
        return query(i<<1,l,mid,t)+query(i<<1|1,mid+1,r,t);
        else if(t==3)
        return max(query(i<<1,l,mid,t),query(i<<1|1,mid+1,r,t));
    }
}
int main(int argc, char const *argv[]) {
    cin>>n>>m;
    built(1,1,n);
    while(m--)
    {
        int t,x,y;
        cin>>t>>x>>y;
        if(t==1)
        {
            updata(1,x,y);
        }
        else
        {
            printf("%d\n",query(1,x,y,t) );
        }
    }
    return 0;
}