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
1 2 3 4
2 1 3
1 4 3
3 1 4
样例输出
6
3
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;
}