您的位置:首页 > 其它

蓝桥杯:操作格子

2014-02-07 20:37 465 查看
问题描述

有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<algorithm>
#include<cstring>
using namespace std;
struct Node{
int left;
int right;
int data;
int sum;
int max;
}a[400000];
void build(int index,int lef,int rig){
a[index].left=lef;
a[index].right=rig;
a[index].max=0;
a[index].data=0;
a[index].sum=0;
if(lef==rig){
return ;
}
int mid=(lef+rig)>>1;
build((index<<1),lef,mid);
build((index<<1)+1,mid+1,rig);
}
void update(int index,int s,int d){
if(a[index].left==a[index].right){a[index].data=d;a[index].max=d;a[index].sum=d;return ;}
int mid=(a[index].left+a[index].right)>>1;
if(mid>=s)
{
update((index<<1),s,d);
}
else if(mid<s)
{
update((index<<1)+1,s,d);
}
a[index].max=(a[(index<<1)].max>a[(index<<1)+1].max)?a[(index<<1)].max:a[(index<<1)+1].max;
a[index].sum=a[(index<<1)].sum+a[(index<<1)+1].sum;
}
int getmax(int index,int s,int e)
{
if(a[index].left==s&&a[index].right==e)return a[index].max;
int mid=(a[index].right+a[index].left)>>1;
if(e<=mid)
{
return getmax((index<<1),s,e);
}
else if(mid<s)
{
return getmax((index<<1)+1,s,e);
}
else
{
return getmax((index<<1),s,mid)>getmax((index<<1)+1,mid+1,e)?getmax((index<<1),s,mid):getmax((index<<1)+1,mid+1,e);
}
}
int getsum(int index,int s,int e)
{
if(a[index].left==s&&a[index].right==e)return a[index].sum;
int mid=(a[index].right+a[index].left)>>1;
if(e<=mid)
{
return getsum((index<<1),s,e);
}
else if(mid<s)
{
return getsum((index<<1)+1,s,e);
}
else
{
return getsum((index<<1),s,mid)+getsum((index<<1)+1,mid+1,e);
}
}
int main(){
int n,m;
while(cin>>n>>m)
{
memset(a,0,sizeof(a));
build(1,1,n);
for(int i=1;i<=n;i++)
{
int t;
cin>>t;
update(1,i,t);
}
for(int i=0;i<m;i++)
{
int op;
cin>>op;
if(op==1)
{
int t,d;
cin>>t>>d;
update(1,t,d);
}
else if(op==2)
{
int s,e;
cin>>s>>e;
cout<<getsum(1,s,e)<<endl;
}
else if(op==3)
{
int s,e;
cin>>s>>e;
cout<<getmax(1,s,e)<<endl;
}
}
}
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  蓝桥杯 线段树