操作格子

问题描述

有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<cmath>
#include<algorithm>
#define MAX 300005
using namespace std;
struct sTree //segment tree
{
    int l, r;
    int sum, max;
};
sTree s[MAX];
void Build(int i, int l, int r)
{
    s[i].l = l;
    s[i].r = r;
    s[i].max = 0;
    s[i].sum = 0;
    if (l == r)return;
    Build(i * 2, l, (l + r) / 2);
    Build(i * 2 + 1, (l + r) / 2 + 1, r);
}
void Insert(int i, int k, int x)
{
    s[i].sum += x;
    if (s[i].max < x)s[i].max = x;
    if (s[i].l == s[i].r)return;
    if (k <= (s[i].l + s[i].r) / 2)
        Insert(i * 2, k, x);
    else
        Insert(i * 2 + 1, k, x);
}
void Change(int i, int k, int x)
{
    if (k == s[i].l && k == s[i].r)
    {
        s[i].sum = x;
        s[i].max = x;
        return;
    }
    int middle = (s[i].l + s[i].r) / 2;
    if (k <= middle)
        Change(i * 2, k, x);
    else
        Change(i * 2 + 1, k, x);
    s[i].sum = s[i * 2].sum + s[i * 2 + 1].sum;
    s[i].max = max(s[i * 2].max, s[i * 2 + 1].max);
}
int QuerySum(int i, int l, int r)
{
    if (l == s[i].l&&r == s[i].r)return s[i].sum;
    int middle = (s[i].l + s[i].r) / 2;
    if (r <= middle)
        return QuerySum(i * 2, l, r);
    else if (l > middle)
        return QuerySum(i * 2 + 1, l, r);
    else return QuerySum(i * 2, l, middle) + QuerySum(i * 2 + 1, middle + 1, r);
}
int QueryMax(int i, int l, int r)
{
    if (l == s[i].l&&r == s[i].r)return s[i].max;
    int middle = (s[i].l + s[i].r) / 2;
    if (r <= middle)
        return QueryMax(i * 2, l, r);
    else if (l > middle)
        return QueryMax(i * 2 + 1, l, r);
    else return max(QueryMax(i * 2, l, middle), QueryMax(i * 2 + 1, middle + 1, r));
}
int main()
{
    int n, m, t, p, x, y;
    cin >> n >> m;
    Build(1, 1, n);
    for (int i = 1; i <= n; i++)
    {
        cin >> t;
        Insert(1, i, t);
    }
    while (m--)
    {
        cin >> p >> x >> y;
        if (p == 1)
            Change(1, x, y);
        else if (p == 2)
            cout << QuerySum(1, x, y) << endl;
        else
            cout << QueryMax(1, x, y) << endl;
    }
    system("pause");
    return 0;
}