操作格子(线段树)

问题描述

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

//C++  265ms	6.902MB
#include <stdio.h>
#define N_MAX 100000
#define L(X) ((X)<<1) 
#define R(X) (((X)<<1)|1)
#define MID(X,Y) (((X)+(Y))>>1) 
#define MAX(X,Y) ((X)>(Y)?(X):(Y))

struct 
{
	int left,right;
	int sum ;
	int m_max ;
}node[N_MAX<<2] ;

void 
Build_Tree ( int const left , int const n , int const right ) 
{
	node[n].left = left ;
	node[n].right = right ;
	node[n].sum = 0 ;
	node[n].m_max = 0 ;
	if ( left < right ) 
	{
		int mid ;
		mid = MID ( left , right ) ;
		Build_Tree ( left , L(n) , mid ) ;
		Build_Tree ( mid + 1 , R(n) , right ) ;
	}
	return ;
}


void 
Travle_Tree ( int const left , int const n , int const right )   //中序遍历线段树
{
	if ( left == right ) 
	{
		printf ("n = %d  sum = %d left = %d right = %d\n" , n , node[n].sum , node[n].left , node[n].right ) ; 
	}
	if ( left < right ) 
	{
		int mid ;
		mid = MID ( left , right ) ;
		Travle_Tree ( left , L(n) , mid ) ;
		printf ("n = %d sum = %d left = %d right = %d\n" , n , node[n].sum , node[n].left , node[n].right ) ; 
		Travle_Tree ( mid+1 , R(n) , right ) ;
	}
	return ; 
}



void	//case 1 
Insert ( int const n , int const index , int const num ) 
{
	if ( node[n].left == node[n].right ) 
	{
		node[n].sum = num ;
		node[n].m_max = num ;
		return ;
	}
	else 
	{
		int mid ;
		mid = MID ( node[n].left , node[n].right ) ;
		if ( index <= mid ) 
		{
			Insert ( L(n) , index , num ) ;
		}
		else 
		{
			Insert ( R(n) , index , num ) ;
		}
		node[n].sum = node[L(n)].sum + node[R(n)].sum ;
		node[n].m_max  = MAX ( node[L(n)].m_max , node[R(n)].m_max ) ;
	}
	return ;
}

int		//case 2 
Find_Sum ( int const n , int const left , int const right ) 
{
	if ( left == node[n].left && right == node[n].right ) 
	{
		return node[n].sum ;
	}
	else 
	{
		int mid ;
		mid = MID ( node[n].left , node[n].right ) ;
		if ( left > mid ) 
		{
			return Find_Sum ( R(n) , left , right ) ;
		}
		else if ( right <= mid ) 
		{
			return Find_Sum ( L(n) , left , right ) ;
		}
		else 
		{
			return Find_Sum ( L(n) ,left , mid ) + Find_Sum ( R(n) , mid + 1 , right ) ;
		}
	}
	return 0 ;
}

int		//case 3
Find_Max ( int const n , int const left , int const right ) 
{
	if ( left == node[n].left && right == node[n].right ) 
	{
		return node[n].m_max ;
	}
	else 
	{
		int mid ;
		mid = MID ( node[n].left , node[n].right ) ;
		if ( left > mid ) 
		{
			return Find_Max ( R(n) , left , right ) ;
		}
		else if ( right <= mid ) 
		{
			return Find_Max ( L(n) , left , right ) ;
		}
		else 
		{
			return MAX ( Find_Max ( L(n) , left , mid ) , Find_Max ( R(n) , mid + 1 , right ) ) ;
		}
	}
	return 0 ;
}

void 
Input ( int const n ) 
{
	int num ;
	for ( int i = 1 ; i <= n ; i ++ ) 
	{
		scanf ("%d" , & num ) ;
		Insert ( 1 , i , num ) ;
	}
	return ; 
}


int 
main ( ) 
{
	int n , m ;
	scanf ("%d%d" , & n , & m ) ;
	Build_Tree (1,1,n );
	Input ( n ) ;
	//Travle_Tree  ( 1 , 1 , n ) ;
	while ( m-- ) 
	{
		int oper ; 
		int left , right ;
		scanf ("%d%d%d" , & oper , &left , &right ) ;
		switch ( oper ) 
		{
		case 1:
			Insert ( 1 , left , right ) ; break ;   // 此时left 为要修改值的下标索引,right 为要修改的值..
		case 2:
			printf ("%d\n" , Find_Sum (1,left , right ) ) ; break ;
		case 3:
			printf ("%d\n" , Find_Max (1,left , right ) ); break ;
		}
	}
	return 0 ;
}