算法训练 操作格子

问题描述

有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>
using namespace std;
#define MAX_NUM 1000000
#define max(x,y) (x)>=(y)?(x):(y)
struct node{
	int left;
	int right;
	int value;
	int sum;
}arr[MAX_NUM];

void BuildTree(int node,int l,int r,int* arrnum){
	if(l==r){
		arr[node].left = l;
		arr[node].right = r;
		arr[node].value = arrnum[l];
		arr[node].sum = arrnum[l];
	}else{
		int mid = (l+r)/2;
		BuildTree(2*node,l,mid,arrnum);
		BuildTree(2*node+1,mid+1,r,arrnum);
		arr[node].value = max(arr[2*node].value,arr[2*node+1].value);
		arr[node].sum = arr[2*node].sum+arr[2*node+1].sum;
		arr[node].left = arr[2*node].left;
		arr[node].right = arr[2*node+1].right;
	}
}

void insert(int node,int x,int y){
	if(arr[node].left==arr[node].right){
		arr[node].value = y;
		arr[node].sum = y;
		return;
	}
	if(x <= (arr[node].left+arr[node].right)/2){
		insert(node*2,x,y);
	}else{
		insert(node*2+1,x,y);
	}
	arr[node].value = max(arr[2*node].value,arr[2*node+1].value);
	arr[node].sum = arr[2*node].sum+arr[2*node+1].sum;
	arr[node].left = arr[2*node].left;
	arr[node].right = arr[2*node+1].right;
}

int Find_Max(int node,int left,int right){
	if(left == arr[node].left && right == arr[node].right){
		return arr[node].value;
	}
	int mid = (arr[node].left + arr[node].right)/2;
	if(left>mid){
		return Find_Max(2*node+1,left,right);
	}else if(right<=mid){
		return Find_Max(2*node,left,right);
	}else{
		return max(Find_Max(2*node,left,mid),Find_Max(2*node+1,mid+1,right));
	}
}

int Find_Sum(int node,int left,int right){
	if(left == arr[node].left && right == arr[node].right){
		return arr[node].sum;
	}
	int mid = (arr[node].left + arr[node].right)/2;
	if(left>mid){
		return Find_Sum(2*node+1,left,right);
	}else if(right<=mid){
		return Find_Sum(2*node,left,right);
	}else{
		return Find_Sum(2*node,left,mid)+Find_Sum(2*node+1,mid+1,right);
	}
}

int main(){
	int n,m;
	cin>>n>>m;
	int* arrnum = new int[n]; 
	for(int i = 1; i<=n; i++){
		cin>>arrnum[i];
	}
	BuildTree(1,1,n,arrnum);//node从1开始 
	while(m--){
		int p,x,y;
		cin>>p>>x>>y;
		switch(p){
			case 1:insert(1,x,y);break;
			case 2:cout<<Find_Sum(1,x,y)<<endl;break;
			case 3:cout<<Find_Max(1,x,y)<<endl;break;
		}	
	}
	return 0;
}