蓝桥-ALGO-8 操作格子

算法训练 操作格子  

问题描述

有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<algorithm>
#define maxn 100005
using namespace std;
struct node
{
    int left,right,maxi;
    long long sum;
}tree[maxn*4];
int an[maxn],m_ans;
long long ans;
void pushup(int i)
{
    tree[i].maxi=max(tree[i<<1].maxi,tree[i<<1|1].maxi);
    tree[i].sum=tree[i<<1].sum+tree[i<<1|1].sum;
}
void build(int i,int left,int right)
{
    tree[i].left=left;
    tree[i].right=right;
    tree[i].maxi=0;
    tree[i].sum=0;
    if(left==right)
    {
        tree[i].sum=an[left];
        tree[i].maxi=an[left];
        return ;
    }
    int mid=(left+right)>>1;
    build(i<<1,left,mid);
    build(i<<1|1,mid+1,right);
    pushup(i);
}
void update(int i,int pos,int data,int left,int right)
{
    if(pos==left&&pos==right)
    {
        tree[i].sum=data;
        tree[i].maxi=data;
        return ;
    }
    int mid=(left+right)>>1;
    if(pos<=mid)
        update(i<<1,pos,data,left,mid);
    else
        update(i<<1|1,pos,data,mid+1,right);
    pushup(i);
}
void query(int i,int ql,int qr,int left,int right)
{
    if(ql<=left&&qr>=right)
    {
        m_ans=max(m_ans,tree[i].maxi);
        ans+=tree[i].sum;
        return ;
    }
    int mid=(left+right)>>1;
    if(ql<=mid)
        query(i<<1,ql,qr,left,mid);
    if(qr>mid)
        query(i<<1|1,ql,qr,mid+1,right);
    pushup(i);
}
int main()
{
    int n,m;

    cin >> n >> m;
    for(int i=1;i<=n;i++)
    {
        cin >> an[i];
    }
    build(1,1,n);
    for(int i=0;i<m;i++)
    {
        int op,x,y;
        cin >> op >> x >> y;
        if(op==1)
        {
            update(1,x,y,1,n);
        }
        else if(op==2)
        {
            ans=0;
            query(1,x,y,1,n);
            cout << ans << endl;
        }
        else if(op==3)
        {
            m_ans=0;
            query(1,x,y,1,n);
            cout << m_ans << endl;
        }
    }
    return 0;
}