线段树操作格子

有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

#include <stdio.h>
#include <stdlib.h>
#include<string.h> 
#define Max 100010
int data[Max],DataSum[Max*4],DataMax[Max*4];
int Index[Max],max,sum;
int GetMax(int a,int b)
{
return (a>b?a:b);
}
void MakeSum(int root,int left,int right,int now,int add)
{
DataSum[root]=DataSum[root]+add;
if(left==right){
return;
}
int mid=(left+right)/2;
if(now<=mid){
MakeSum(root*2, left, mid, now, add);
}
if(now>=mid+1){
MakeSum(root*2+1, mid+1, right, now, add);
}
return;

/*DataSum[root] = add;
root/=2;
while(root>=1)
{
DataSum[root] =DataSum[root*2]+DataSum[root*2+1]; 
root/=2;
}*/
}
void MakeMax(int root,int now)
{
DataMax[root] = now;
root/=2;
while(root>=1)
{
DataMax[root] = GetMax(DataMax[root*2+1],DataMax[root*2]);
root/=2;
}
}
void Init(int root,int start,int end)
{
if(start>=end)
{
Index[start]=root;
return; 
} 
int mid = (start+end)/2;
Init(root*2,start,mid);
Init(root*2+1,mid+1,end);
}
void PrintSum(int root,int left,int right,int start,int end)
{
int mid;
if(start >= left&&end <= right)//坑 
{
sum += DataSum[root];
return ;
}
mid = (start+end)/2; 
if(mid>=left){
PrintSum(root*2, left, right, start, mid);
}
if(mid+1<=right){
PrintSum(root*2+1, left, right, mid+1, end);
}
return;
}
void PrintMax(int root,int left,int right,int start,int end)
{
int mid;
if(start >= left&&end <= right)//坑 
{
max = GetMax(DataMax[root],max);
return; 
}
mid = (start+end)/2; 
if(mid>=left){
PrintMax(root*2, left, right, start, mid);
}
if(mid+1<=right){
PrintMax(root*2+1, left, right, mid+1, end);
}
return;
}
int main()
{
memset(DataSum,0,sizeof(DataSum));
memset(DataMax,-1,sizeof(DataMax));
memset(Index,0,sizeof(int)*Max); 
int i,N,K,p,x,y;
scanf("%d%d",&N,&K);
for(i=1;i<=N;i++)
scanf("%d",&data[i]); 
Init(1,1,N);
for(i=1;i<=N;i++)
{
MakeSum(1,1,N,i,data[i]);
MakeMax(Index[i],data[i]);
}
for(i=0;i<K;i++)
{
scanf("%d%d%d",&p,&x,&y);
if(p == 1)
{
MakeSum(1,1,N,x,y-data[x]);
data[x] = y;
MakeMax(Index[x],y);
}
if(p == 2)
{
sum = 0;
PrintSum(1,x,y,1,N);
printf("%d\n",sum);
}
if(p == 3)
{
max = -1;
PrintMax(1,x,y,1,N);
printf("%d\n",max); 
} 
} 
return 0;
}