59-算法训练 操作格子 （线段树）

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

　　　　　　　　　　　　　　算法训练 操作格子  

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

 

思路：线段树的裸题吧，注意数据的大小。

#include <iostream>
#include <cstring>
using namespace std;
typedef struct node{
	int l, r;        //作为节点的区间 
	long long v, s;  //v记录最大值，s记录区间和 
}Node;
Node t[400000];
long long n, m, ans = 0, Max = -0x3f3f3f3f;

void build(int l, int r, int k){
	t[k].l = l, t[k].r = r;
	if(l == r){
		cin >> t[k].v;
		t[k].s = t[k].v;
		return ;           //每个查找完都要结束
	}
	int mid = (l + r) / 2;
	build(l, mid, k * 2);
	build(mid + 1, r, k * 2 + 1);
	t[k].v = max(t[k * 2].v, t[k * 2 + 1].v);
	t[k].s = t[k * 2].s + t[k * 2 + 1].s;
}

void update(int c, int v, int k){
	if(t[k].l == t[k].r && t[k].l == c){
		t[k].v = v;
		t[k].s = v;   //这个也要改才行哇！！！ 
		return ;　　　 //每个查找玩，都要结束　　
	}
	int mid = (t[k].l + t[k].r) / 2;
	if(c <= mid){
		update(c, v, k * 2);
	}
	else{
		update(c, v, k * 2 + 1);
	}
	t[k].v = max(t[k * 2].v, t[k * 2 + 1].v);
	t[k].s = t[k * 2].s + t[k * 2 + 1].s;
}

void query(int l, int r, int k){
	if(t[k].r < l || t[k].l > r){
		return ;
	}
	if(l <= t[k].l && r >= t[k].r){
		ans += t[k].s;
		if(Max < t[k].v){
			Max = t[k].v;
		}
		return ;    //查找玩要立即结束
	}
	int mid = (t[k].l + t[k].r) / 2; //这里是节点的区间中点
	if(mid > r){
		query(l, r, k * 2);
	}
	else if(mid < l){
		query(l, r, k * 2 + 1);
	}
	else{
		query(l, mid, k * 2);
		query(mid + 1, r, k * 2 + 1);
	}
}

int main(){
	cin >> n >> m;
	build(1, n, 1);
	while(m--){
		int i, x, y;
		Max = -0x3f3f3f3f;  //每次都要初始化
		ans = 0; 
		cin >> i >> x >> y;
		if(i == 1){
			update(x, y, 1);
		}
		else if(i == 2){
			query(x, y, 1);
			cout << ans << endl;
		}
		else if(i == 3){
			query(x, y, 1);
			cout << Max << endl;
		}
	}
	return 0;
}