问题描述
有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。
很赤裸的线段树,没有什么技巧好言,线段树+RMQ
不过我写得很挫,特别是change函数里,wa了一天。
后来参照lrj的白皮书回溯算法。才A的。。
很惭愧。。
#include<cstdio>
#include<cstring>
int s[100005],ma[600005];
int sum[600005];
int dp[100003][20];
int max(int a,int b)
{
return a>b?a:b;
}
void Init(int n)
{
int i,j;
for(i=1;i<=n;i++)
dp[i][0] = s[i]-s[i-1];
for(j=1;(1<<j)<=n+1;j++)
for(i=0;i+(1<<j)<=n+1;i++)
dp[i][j] = max(dp[i][j-1],dp[i + (1<<(j-1))][j-1]);
}
int RMQ(int l,int r)
{
int k=0;
while((1<<(k+1)) <= r-l+1) k++;
return max(dp[l][k],dp[r-(1<<k)+1][k]);
}
void build_tree(int l,int r,int o)
{
if(l>r) return;
sum[o] = s[r]-s[l-1];
ma[o] = RMQ(l,r);
if(l==r) return;
int m = (l+r)/2;
build_tree(l,m,o*2);
build_tree(m+1,r,o*2+1);
}
void change(int v2,int v3,int o,int l,int r)
{
int m = (l+r)/2;
if(l==r)
{
ma[o] = v3;
sum[o] = v3;
}
else
{
if(v2<=m) change(v2,v3,o*2,l,m);
else
change(v2,v3,o*2+1,m+1,r);
ma[o] = max(ma[o*2],ma[o*2+1]);
sum[o] = sum[o*2]+sum[o*2+1];
}
}
int seeksum(int v2,int v3,int o,int l,int r)
{
if(l==v2 && r==v3)
return sum[o];
int m = (l+r)/2;
if(v2>m)
return seeksum(v2,v3,o*2+1,m+1,r);
else if(v3<=m)
return seeksum(v2,v3,o*2,l,m);
else
return seeksum(v2,m,o*2,l,m)+seeksum(m+1,v3,o*2+1,m+1,r);
}
int seekmax(int v2,int v3,int o,int l,int r)
{
if(l==v2 && r==v3)
return ma[o];
int m = (l+r)/2;
if(v2>m)
return seekmax(v2,v3,o*2+1,m+1,r);
else if(v3<=m)
return seekmax(v2,v3,o*2,l,m);
else
return max(seekmax(v2,m,o*2,l,m),seekmax(m+1,v3,o*2+1,m+1,r));
}
int main()
{
int n,m;
while(scanf("%d%d",&n,&m)!=EOF)
{
s[0] = 0;
for(int i=1;i<=n;i++)
{
scanf("%d",&s[i]);
s[i] = s[i]+s[i-1];
}
Init(n);
build_tree(1,n,1);
int v1,v2,v3;
for(int j=0;j<m;j++)
{
scanf("%d%d%d",&v1,&v2,&v3);
switch(v1)
{
case 1:change(v2,v3,1,1,n);break;
case 2:printf("%d\n",seeksum(v2,v3,1,1,n));break;
case 3:printf("%d\n",seekmax(v2,v3,1,1,n));break;
}
}
}
return 0;
}