操作格子(线段树)
扫描二维码
随时随地手机看文章
有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 31 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。
自己感觉这个问题有点像士兵杀敌问题,当初自己想到了树状数组,但是在求最值问题似乎不能成功,后来只能用到线段树,代码自己不会写,java的更不会,先用C语言尝试了下,做了半天没有成功,编译成功的代码是基于数据结构专题--线段树,感谢你。
#include
#include
using namespace std;
#define lson l , m , rt << 1
#define rson m + 1 , r , rt << 1 | 1
const int maxn = 4000000;
int MAX[maxn<<2];
int sum[maxn<<2];
void PushUP(int rt)
{
MAX[rt] = max(MAX[rt<<1] , MAX[rt<<1|1]);
sum[rt] = sum[rt<<1] + sum[rt<<1|1];
}
void build(int l,int r,int rt) {
if (l == r) {
scanf("%d",&MAX[rt]);
sum[rt]=MAX[rt];
return ;
}
int m = (l + r) >> 1;
build(lson);
build(rson);
PushUP(rt);
}
void update(int p,int sc,int l,int r,int rt)
{
if (l == r)
{
MAX[rt] = sc;
sum[rt] = sc;
return ;
}
int m = (l + r) >> 1;
if (p <= m)
update(p , sc , lson);
else
update(p , sc , rson);
PushUP(rt);
}
int query(int L,int R,int l,int r,int rt) {
if (L <= l && r <= R) {
return MAX[rt];
}
int m = (l + r) >> 1;
int ret = 0;
if (L <= m) ret = max(ret , query(L , R , lson));
if (R > m) ret = max(ret , query(L , R , rson));
return ret;
}
int querysum(int L,int R,int l,int r,int rt)
{
if( L <= l && r <= R )
{
return sum[rt];
}
int m = ( l + r ) >> 1;
int ans=0;
if(L<=m )
ans+=querysum(L,R,lson);
if(R>m)
ans+=querysum(L,R,rson);
return ans;
}
int main()
{
int n , m;
int s_val,ss_val;
FILE *fp;
scanf("%d%d",&n,&m);
{
build(1 , n , 1);
while (m --)
{
int op;
int a , b;
scanf("%d %d %d",&op,&a,&b);
if (op==1)
{
update(a , b , 1 , n , 1);
}
else if(op==2)
{
printf("%dn",querysum(a,b,1,n,1));
}
else if(op==3)
{
printf("%dn",query(a,b,1,n,1));
}
}
}
return 0;
}