题目描述
很多学校流行一种比较的习惯。老师们很喜欢询问,从某某到某某当中,分数最高的是多少。
这让很多学生很反感。
不管你喜不喜欢,现在需要你做的是,就是按照老师的要求,写一个程序,模拟老师的询问。当然,老师有时候需要更新某位同学的成绩。
Input
本题目包含多组测试,请处理到文件结束。
在每个测试的第一行,有两个正整数 N 和 M ( 0<N<=200000,0<M<5000 ),分别代表学生的数目和操作的数目。
学生ID编号分别从1编到N。
第二行包含N个整数,代表这N个学生的初始成绩,其中第i个数代表ID为i的学生的成绩。
接下来有M行。每一行有一个字符 C (只取’Q’或’U’) ,和两个正整数A,B。
当C为’Q’的时候,表示这是一条询问操作,它询问ID从A到B(包括A,B)的学生当中,成绩最高的是多少。
当C为’U’的时候,表示这是一条更新操作,要求把ID为A的学生的成绩更改为B。
Output
对于每一次询问操作,在一行里面输出最高成绩。
Sample Input
5 6
1 2 3 4 5
Q 1 5
U 3 6
Q 3 4
Q 4 5
U 2 9
Q 1 5
Sample Output
5
6
5
9
思路
这题就是裸线段树能解决的问题,查询时根据最初划定的区间一直往下二分查找直到查找区间的左右界(会随着查找过程而改变)与当前所在节点的左右界完全相同时返回当前节点的最大值即可,增加一个记录第i个学生的成绩最终在线段树中的位置的数组c,更新成绩时先更新线段树中这个节点的最大值,再一路更新父节点的最大值直到树根。
特别注意!
在查找函数中不要用define max(a,b) ((a)>(b)?(a):(b))
的形式来刷新最大值!因为当写下max(search(l1,r1),search(l2,r2))
时代码会被展开成(search(l1,r1)>search(l2,r2)?search(l1,r1):search(l2,r2))
也就是说search(l1,r1)
或search(l2,r2)
会被执行两次,会严重影响程序效率!附上代码。
TLE版本
#include<stdio.h>
#define max(a,b) ((a)>(b)?(a):(b))
struct st{
int l,r,ma;
};
struct st dat[800020];
int c[200020];
int ad[200020];
char ch[3];
int n,m,a,b;
void mt(int now,int l,int r)
{
int mid=(l+r)>>1;
dat[now].l=l;dat[now].r=r;
if(l==r)
{
dat[now].ma=c[l];
ad[l]=now;
return;
}else
{
mt(now*2,l,mid);
mt(now*2+1,mid+1,r);
dat[now].ma=max(dat[now*2].ma,dat[now*2+1].ma);
return;
}
}
int sear(int now,int l,int r)
{
int mid=(dat[now].l+dat[now].r)>>1;
if(l==dat[now].l&&r==dat[now].r)return dat[now].ma;
if(r<=mid) return sear(now*2,l,r);
if(l>mid) return sear(now*2+1,l,r);
return max(sear(now*2,l,mid),sear(now*2+1,mid+1,r));
}
void ed(int now)
{
if(!now) return;
int s=max(dat[now*2].ma,dat[now*2+1].ma);
if(s!=dat[now].ma)
{
dat[now].ma=s;
ed(now/2);
}
return;
}
int main()
{
while(scanf("%d%d",&n,&m)!=EOF)
{
for(int i=1;i<=n;i++)
scanf("%d",&c[i]);
mt(1,1,n);
for(int i=1;i<=m;i++)
{
scanf("%s",ch);
scanf("%d%d",&a,&b);
if(ch[0]=='Q') printf("%d\n",sear(1,a,b));
else
{
dat[ad[a]].ma=b;ed(ad[a]/2);
}
}
}
return 0;
}
AC版本
#include<stdio.h>
#define max(a,b) ((a)>(b)?(a):(b))
struct st{
int l,r,ma;
};
struct st dat[800020];
int c[200020];
int ad[200020];
char ch[3];
int n,m,a,b;
void mt(int now,int l,int r)
{
int mid=(l+r)>>1;
dat[now].l=l;dat[now].r=r;
if(l==r)
{
dat[now].ma=c[l];
ad[l]=now;
return;
}else
{
mt(now*2,l,mid);
mt(now*2+1,mid+1,r);
dat[now].ma=max(dat[now*2].ma,dat[now*2+1].ma);
return;
}
}
int sear(int now,int l,int r)
{
int mid=(dat[now].l+dat[now].r)>>1;
if(l==dat[now].l&&r==dat[now].r)return dat[now].ma;
if(r<=mid) return sear(now*2,l,r);
if(l>mid) return sear(now*2+1,l,r);
int temp1=sear(now*2,l,mid),temp2=sear(now*2+1,mid+1,r);
return max(temp1,temp2);
}
void ed(int now)
{
if(!now) return;
int s=max(dat[now*2].ma,dat[now*2+1].ma);
if(s!=dat[now].ma)
{
dat[now].ma=s;
ed(now/2);
}
return;
}
int main()
{
while(scanf("%d%d",&n,&m)!=EOF)
{
for(int i=1;i<=n;i++)
scanf("%d",&c[i]);
mt(1,1,n);
for(int i=1;i<=m;i++)
{
scanf("%s",ch);
scanf("%d%d",&a,&b);
if(ch[0]=='Q') printf("%d\n",sear(1,a,b));
else
{
dat[ad[a]].ma=b;ed(ad[a]/2);
}
}
}
return 0;
}