算法训练 操作格子
时间限制：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
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<cstring> #include<string> #include<cstdio> #include<cmath> using namespace std; struct node { int left,right,n,sum; }tree[1000000*4]; void init(int l,int r,int i) { tree[i].left = l; tree[i].right = r; tree[i].n = 0; tree[i].sum = 0; if(l==r) return ; else { int mid = (l + r) / 2; init(l,mid,2*i); init(mid+1,r,2*i+1); } } void insert(int i,int x,int m) { if(x>=tree[i].left && x<=tree[i].right) { tree[i].n = m; tree[i].sum = m; } if(tree[i].left == tree[i].right) return ; int mid = (tree[i].left + tree[i].right) / 2; if(x>mid) insert(2*i+1,x,m); else insert(2*i,x,m); tree[i].sum = tree[2*i].sum + tree[2*i+1].sum; tree[i].n = max(tree[2*i].n ,tree[2*i+1].n); } int findmax(int x,int y,int i) { if(x==tree[i].left && y==tree[i].right) return tree[i].n; int mid = (tree[i].left + tree[i].right) / 2; if(x>mid) return findmax(x,y,2*i+1); else if(y<=mid) return findmax(x,y,2*i); else return max(findmax(x,mid,2*i),findmax(mid+1,y,2*i+1)); } int findsum(int x,int y,int i) { if(x==tree[i].left && y==tree[i].right) return tree[i].sum; int mid = (tree[i].left + tree[i].right) / 2; if(x > mid) return findsum(x,y,2*i+1); else if(y<=mid) return findsum(x,y,2*i); else return findsum(x,mid,2*i)+findsum(mid+1,y,2*i+1); } int main() { int n,m,i,b,x,y,cas; cin>>n>>m; init(1,n,1); for(i=1;i<=n;i++) { cin>>b; insert(1,i,b); } while(m--) { cin>>cas>>x>>y; if(cas==1) insert(1,x,y); else if(cas==2) cout<<findsum(x,y,1)<<endl; else if(cas==3) cout<<findmax(x,y,1)<<endl; } system("pause"); return 0; }