问题描述
有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
1 2 3 4
2 1 3
1 4 3
3 1 4
样例输出
6
3
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; }