蓝桥杯 算法训练 操作格子 (线段树)

题目链接：



http://lx.lanqiao.org/problem.page?gpid=T18

问题描述

有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。

分析：

      本题是简单的线段树，本题主要是既要求线段和，又要求线段最大值。。 不过可以分开考虑。问题就变得很简单了。

如果还不了解线段树，可以看看，有详细介绍注释（我也是菜鸟）

 求线段最大值：http://blog.csdn.net/qq_33850438/article/details/50650900

 求线段和        ：http://blog.csdn.net/qq_33850438/article/details/50604731

代码（AC）

#include"stdlib.h"
#include"stdio.h"
#include"algorithm"
#include"ctype.h"
#include"iostream"
using namespace std;
const int maxn=1005;
const int INF=-1e9;
int a[maxn];
long long sum[maxn*2],maxx[maxn*2];//查询线段和、最大值 
int n,m;
void Build(int node,int start,int end)
{//建立sum maxx
   if(start==end) 
   {
      sum[node]=a[start];
      maxx[node]=a[start];
   }
   else
    {
    	int m=start+(end-start)/2;
        Build(node*2,start,m);
		Build(node*2+1,m+1,end);
		sum[node]=sum[node*2]+sum[node*2+1];  
		maxx[node]=max(maxx[node*2],maxx[node*2+1]); 	
	}	
}
long long Query1(int node,int start,int end,int le,int ri)
{//查询连续和 
	if(le<=start&&end<=ri) return sum[node];
	int m=start+(end-start)/2;
	long long L=0,R=0;
	if(m>=le) L=Query1(node*2,start,m,le,ri);
	if(ri>m)  R=Query1(node*2+1,m+1,end,le,ri);
	return L+R;
}
long long Query2(int node,int start,int end,int le,int ri)
{//查询最大值 
	if(le<=start&&end<=ri) return maxx[node];
	int m=start+(end-start)/2;
	long long ans=INF;
	if(m>=le) ans=max(ans,Query2(node*2,start,m,le,ri));
	if(ri>m)  ans=max(ans,Query2(node*2+1,m+1,end,le,ri));
	return ans;
}
void Update(int node,int start,int end,int p,int v)  
//修改a[p]=v；
{
    if(start==end)
	{
	 	sum[node]=v;
	 	maxx[node]=v;
	}
	else
	{
	    int m=start+(end-start)/2;
		if(p<=m)
		   Update(node*2,start,m,p,v);
		else
		   Update(node*2+1,m+1,end,p,v);
		sum[node]=sum[node*2]+sum[node*2+1];
		maxx[node]=max(maxx[node*2],maxx[node*2+1]);
	}	
}
void Solve()
{
   scanf("%d%d",&n,&m);
   for(int i=0;i<n;i++)
   	  scanf("%d",a+i);
	Build(1,0,n-1);
	 while(m--)
	 {
	 	//for(int i=0;i<n*2;i++)
	    //   printf("%d ",maxx[i]);
	 	int p,x,y;
	 	long long ans=0;
	 	scanf("%d%d%d",&p,&x,&y);
	 	if(p==1)
	 		Update(1,1,n,x,y);
	 	else if(p==2)
	 	    {
			   ans=Query1(1,1,n,x,y);
			   printf("%lld\n",ans);
		    }
	 	    else
	 	    {
	 	    	ans=Query2(1,1,n,x,y);
	 	    	printf("%lld\n",ans);
			}
	 }	
} 
int main()
{
        Solve();
	return 0;
}