蓝桥网 算法训练 操作格子

问题描述

有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

这就是一道线段树的最基础的模板题

AC代码：

# include <stdio.h>
# include <iostream>
# include <string.h>
# include <algorithm>
using namespace std;
typedef long long int ll;
int a[100010], r1[4*100010], r2[4*100010];
int ans1, ans2, index[4*100010];
void update1(int root, int l, int r, int cur, int add){
r1[root]=r1[root]+add;
if(l==r){
return;
}
int mid=(l+r)/2;
if(cur<=mid){
update1(root*2, l, mid, cur, add);
}
if(cur>=mid+1){
update1(root*2+1, mid+1, r, cur, add);
}
return;
}
void update2(int k, int v){
r2[k]=v;
k=k/2;
while(k>0){
r2[k]=max(r2[k*2], r2[2*k+1]);
k=k/2;
}
return;
}
void query1(int root, int l, int r, int L, int R){
if(l>=L&&R>=r){
ans1=ans1+r1[root];
return;
}
int mid=(l+r)/2;
if(mid>=L){
query1(root*2, l, mid, L, R);
}
if(mid+1<=R){
query1(root*2+1, mid+1, r, L, R);
}
return;
}
void init(int root, int l, int r){
if(l==r){
index[l]=root;
return;
}
int mid=(l+r)/2;
init(root*2, l, mid);
init(root*2+1, mid+1, r);
}
void query2(int root, int l, int r, int L, int R){
if(l>=L&&R>=r){
if(r2[root]>ans2){
ans2=r2[root];
}
return;
}
int mid=(l+r)/2;
if(mid>=L){
query2(root*2, l, mid, L, R);
}
if(mid+1<=R){
query2(root*2+1, mid+1, r, L, R);
}
return;
}
int main(){
int i, j, k;
int m, n, x, y, p;
scanf("%d%d", &n, &m);
for(i=1; i<=n; i++){
scanf("%d", &a[i]);
}
memset(r2, -1, sizeof(r2));
init(1, 1, n);
for(i=1; i<=n; i++){
update1(1, 1, n, i, a[i]);
update2(index[i], a[i]);
}
for(i=1; i<=m; i++){
scanf("%d%d%d", &p, &x, &y);
if(p==1){
update1(1, 1, n, x, y-a[x]);
a[x]=y;
update2(index[x], y);
}
else{
if(p==2){
ans1=0;
query1(1, 1, n, x, y);
printf("%d\n", ans1);
}
else{
ans2=-1;
query2(1, 1, n, x, y);
printf("%d\n", ans2);
}
}
}
return 0;
}