蓝桥杯 算法训练 操作格子 （线段树）

算法训练 操作格子  

时间限制：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

样例输出

6
3

数据规模与约定

对于20%的数据n <= 100，m <= 200。

对于50%的数据n <= 5000，m <= 5000。

对于100%的数据1 <= n <= 100000，m <= 100000，0 <= 格子权值 <= 10000。

就是简单的pushup大法

#include <stdio.h>
#define MAX 500000
struct segtree{
	int left,right,sum,max;
}st[MAX];

int fmax(int a , int b)
{
	return a>b?a:b ;
}

void pushUp(int pos)
{
	st[pos].sum = st[pos<<1].sum + st[pos<<1|1].sum ;
	st[pos].max = fmax(st[pos<<1].max , st[pos<<1|1].max);
}

void creat(int left , int right ,int pos)
{
	st[pos].sum = 0 ;
	st[pos].max = -1 ;
	st[pos].left = left ;
	st[pos].right = right ;
	if(left == right)
	{
		scanf("%d",&st[pos].sum) ;
		st[pos].max = st[pos].sum ;
		return ;
	}
	int mid = (left+right)>>1 ;
	creat(left,mid,pos<<1) ;
	creat(mid+1,right,pos<<1|1) ;
	pushUp(pos);
}

void update(int rt,int data ,int pos)
{
	if(rt == st[pos].left && st[pos].left == st[pos].right)
	{
		st[pos].sum = data;
		st[pos].max = data ;
		return ;
	}
	int mid = (st[pos].left + st[pos].right)>>1;
	if(rt>mid)
	{
		update(rt,data,pos<<1|1);
	}
	else
	{
		update(rt,data,pos<<1);
	}
	pushUp(pos);
}

int query(int left , int right, int pos)
{
	if(left == st[pos].left && right == st[pos].right)
	{
		return st[pos].sum;
	}
	int mid = (st[pos].left+st[pos].right)>>1 ,sum = 0 ;
	if(left > mid)
	{
		sum += query(left,right,pos<<1|1) ;
	}
	else if(right <= mid)
	{
		sum += query(left,right,pos<<1) ;
	}
	else
	{
		sum += query(left,mid,pos<<1) ;
		sum += query(mid+1,right,pos<<1|1) ;
	}
	return sum ;
}

int findMax(int left , int right ,int pos)
{
	if(left == st[pos].left && right == st[pos].right)
	{	
		return st[pos].max ;
	}
	int mid = (st[pos].left+st[pos].right)>>1;
	if(left > mid)
	{
		return findMax(left,right,pos<<1|1);
		
	}
	else if(right <= mid)
	{
		return findMax(left,right,pos<<1) ;
	}
	else
	{
		return fmax(findMax(left,mid,pos<<1),findMax(mid+1,right,pos<<1|1));
	}
	
}
int main()
{
	int n , m;
	while(scanf("%d%d",&n,&m) != EOF)
	{
		int p,x,y;
		creat(1,n,1);
		for(int i = 0 ; i < m ; ++i)
		{
			scanf("%d%d%d",&p,&x,&y) ;
			switch(p)
			{
				case 1 :
					{
						update(x,y,1);
					}
					break;
				case 2:
					{
						if(x>y)
						{
							int t = x;x = y ;y = t;
						}
						
						printf("%d\n",query(x,y,1));
					}break ;
				case 3:
					{
						if(x>y)
						{
							int t = x;x = y ;y = t;
						}
						
						printf("%d\n",findMax(x,y,1));
					}	
					default:break;
			}
		}
		
	}
	
	return 0 ;
}