蓝桥杯 操作格子（线段树）

问题描述

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

很赤裸的线段树，没有什么技巧好言，线段树+RMQ

不过我写得很挫，特别是change函数里，wa了一天。

后来参照lrj的白皮书回溯算法。才A的。。

很惭愧。。

#include<cstdio>
#include<cstring>
int s[100005],ma[600005];
int sum[600005];
int dp[100003][20];
int max(int a,int b)
{
	return a>b?a:b;
}
void Init(int n)
{
	int i,j;
	for(i=1;i<=n;i++)
		dp[i][0] = s[i]-s[i-1];
	for(j=1;(1<<j)<=n+1;j++)
		for(i=0;i+(1<<j)<=n+1;i++)
			dp[i][j] = max(dp[i][j-1],dp[i + (1<<(j-1))][j-1]);
		
}
int RMQ(int l,int r)
{
	int k=0;
	while((1<<(k+1)) <= r-l+1) k++;
	return max(dp[l][k],dp[r-(1<<k)+1][k]);
}
void build_tree(int l,int r,int o)
{
	if(l>r) return;
	sum[o] = s[r]-s[l-1];
	ma[o] = RMQ(l,r);
	if(l==r) return;
	int m = (l+r)/2;
	build_tree(l,m,o*2);
	build_tree(m+1,r,o*2+1);
}
void change(int v2,int v3,int o,int l,int r)
{
	int m = (l+r)/2;
	if(l==r)
	{
		ma[o] = v3;
		sum[o] = v3;
	}
	else
	{
		if(v2<=m) change(v2,v3,o*2,l,m);
		else
			change(v2,v3,o*2+1,m+1,r);
		ma[o] = max(ma[o*2],ma[o*2+1]);
		sum[o] = sum[o*2]+sum[o*2+1];
	}
	
}
int seeksum(int v2,int v3,int o,int l,int r)
{
	if(l==v2 && r==v3)
		return sum[o];
	int m = (l+r)/2;
	if(v2>m)
		return seeksum(v2,v3,o*2+1,m+1,r);
	else if(v3<=m)
		return seeksum(v2,v3,o*2,l,m);
	else
		return seeksum(v2,m,o*2,l,m)+seeksum(m+1,v3,o*2+1,m+1,r);
	
}
int seekmax(int v2,int v3,int o,int l,int r)
{
	if(l==v2 && r==v3)
		return ma[o];
	int m = (l+r)/2;
	if(v2>m)
		return seekmax(v2,v3,o*2+1,m+1,r);
	else if(v3<=m)
		return seekmax(v2,v3,o*2,l,m);
	else
		return max(seekmax(v2,m,o*2,l,m),seekmax(m+1,v3,o*2+1,m+1,r));
}
int main()
{
	int n,m;
	while(scanf("%d%d",&n,&m)!=EOF)
	{
		s[0] = 0;
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&s[i]);
			s[i] = s[i]+s[i-1];
		}
		Init(n);
		build_tree(1,n,1);
		int v1,v2,v3;
		for(int j=0;j<m;j++)
		{
			scanf("%d%d%d",&v1,&v2,&v3);
			switch(v1)
			{
			case 1:change(v2,v3,1,1,n);break;
			case 2:printf("%d\n",seeksum(v2,v3,1,1,n));break;
			case 3:printf("%d\n",seekmax(v2,v3,1,1,n));break;
			}
		}
		
		
	}
	return 0;
}