hdu 1754:I Hate It(线段树(树状数组)应用之:区间最值问题)
题目链接:hdu 1754
Problem Description
很多学校流行一种比较的习惯。老师们很喜欢询问,从某某到某某当中,分数最高的是多少。
这让很多学生很反感。
不管你喜不喜欢,现在需要你做的是,就是按照老师的要求,写一个程序,模拟老师的询问。当然,老师有时候需要更新某位同学的成绩。
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
Hint
Huge input,the C function scanf() will work better than cin
题解:线段树区间最值模板题之一:点修改,及区间查询
代码如下:
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
using namespace std;
const int INF=0xffffff0; //设一个最大值
const int maxn=200010; //学生数目(即叶子结点数目)
/**结点结构*/
struct Node
{
int L,R;
int v;
};
Node tree[4*maxn]; //用数组代替指针表示线段树
int a[maxn]; //存放学生信息
int maxV=-INF; //表示最大值
/**建立线段树,并附初值*/
void build_Tree(int root,int L,int R)
{
tree[root].L=L;
tree[root].R=R;
if(L==R)
{
tree[root].v=a[L];
return ;
}
int mid=(tree[root].L+tree[root].R)/2;
/**向下直到叶子结点为止*/
build_Tree(2*root,L,mid);
build_Tree(2*root+1,mid+1,R);
tree[root].v=max(tree[2*root].v,tree[2*root+1].v); //回溯时获得左右孩子最大值
}
/**更新操作(即线段树维护)*/
void update(int root,int i,int v)
{
if(tree[root].L==i&&tree[root].R==i)
{
tree[root].v=v;
return ;
}
int mid=(tree[root].L+tree[root].R)/2;
if(i<=mid)
update(2*root,i,v);
else
update(2*root+1,i,v);
tree[root].v=max(tree[2*root].v,tree[2*root+1].v); //回溯获得左右孩子最大值
}
/**查询区间[s,e]中的最大值*/
void Query(int root,int s,int e)
{
if(tree[root].L==s&&tree[root].R==e)
{
maxV=max(maxV,tree[root].v);
return ;
}
int mid=(tree[root].L+tree[root].R)/2;
/**区间过大*/
if(e<=mid)
Query(2*root,s,e);
else if(s>mid)
Query(2*root+1,s,e);
else
{
Query(2*root,s,mid);
Query(2*root+1,mid+1,e);
}
}
int main()
{
int n,m;
while(scanf("%d%d",&n,&m)!=EOF)
{
memset(tree,-INF,sizeof(tree));
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
build_Tree(1,1,n);
char ch; //表示是询问操作还是更新操作
int a,b; //如果是询问操作的话,则表示区间[a,b];如果是更新操作,则表示学生a成绩变为b
for(int i=1;i<=m;i++)
{
getchar();
scanf("%c",&ch);
scanf("%d%d",&a,&b);
if(ch=='Q')
{
maxV=-INF;
Query(1,a,b);
printf("%d\n",maxV);
}
else
update(1,a,b);
}
}
return 0;
}
题解二:树状数组求解
虽然我觉得这种最值问题,不像求和问题,树状数组处理似乎看上去并不怎么好,但是为了复习树状数组,还是用树状数组做了下:
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
int a[200010];
int h[200010];
int n,m;
inline int lowbit(int x)
{
return x&(-x);
}
void update(int x)
{
int lx;
while(x<=n)
{
h[x]=a[x];
lx=lowbit(x);
for(int i=1;i<lx;i<<=1)
h[x]=max(h[x],h[x-i]);
x+=lowbit(x);
}
}
int query(int x,int y)
{
int ans=0;
while(y>=x)
{
ans=max(a[y],ans);
y--;
for(;y-lowbit(y)>=x;y-=lowbit(y))
ans=max(h[y],ans);
}
return ans;
}
int main()
{
while(~scanf("%d%d",&n,&m))
{
memset(a,0,sizeof(a));
memset(h,0,sizeof(h));
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
update(i);
}
char ch;
int x,y;
while(m--)
{
getchar();
scanf("%c",&ch);
scanf("%d%d",&x,&y);
if(ch=='U')
{
a[x]=y;
update(x);
}
else
{
printf("%d\n",query(x,y));
}
}
}
return 0;
}