HDU-1754 I Hate It(线段树 单点修改 找最大值)
Time Limit: 9000/3000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)
这让很多学生很反感。
不管你喜不喜欢,现在需要你做的是,就是按照老师的要求,写一个程序,模拟老师的询问。当然,老师有时候需要更新某位同学的成绩。
在每个测试的第一行,有两个正整数 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。
暴力 O(n^2)
线段树 O(nlogn)
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <algorithm>
using namespace std;
typedef struct segmentree
{
int left;
int right;
long long sum; //long long 型
int maxn;
}segmentree;
//线段树是满二叉树形式,但不是树的链表结构
//所以可以用下标法找结点
segmentree segtree[800005];
int a[800005];
void push_up(int root)
{
segtree[root].maxn = max(segtree[root<<1].maxn ,segtree[root<<1|1].maxn);
}
void CreateSegmentree(int root,int l,int r) //创建线段树
{
segtree[root].left = l;
segtree[root].right = r;
if(l == r)
{
segtree[root].maxn= a[l];
}
else
{
int mid = (l+r)>>1;//mid=(l+r)>>1
CreateSegmentree(root<<1,l,mid);
CreateSegmentree(root<<1|1,mid+1,r); //右移|1才是加1的意思
push_up(root);
//递归的终止就是l==r
}
return;
}
int query(int root,int l,int r) //查询线段树的各个区间。
{
if(segtree[root].left>r||segtree[root].right<l)
return 0; //必须先在这个区间里,不是就返回0;
else if(segtree[root].left==l&&segtree[root].right==r)
return segtree[root].maxn;
int mid=(segtree[root].left + segtree[root].right) >> 1; //mid是构树的关键,也是查询的关键
if(l > mid) //l大于mid,就进入右子树,因为mid在左子树里
return query(root<<1|1,l,r);
else if(r <= mid) //r小于等于mid,就进入左子树 ,因为mid不管n是奇数还是偶数,mid都是在左子树里的
return query(root<<1,l,r);
//以上为两种边界情况:左区间都是在右子树,右区间都是在左子树
else //区间在左右子树之间。 以mid为界,将区间拆分为别在左子树和右子树的两个区间。
return max(query(root<<1,l,mid),query(root<<1|1,mid + 1,r)); //区间是三种情况,所以要这样判断
}
void revise(int root,int x, int val) //单点修改,单点修改判断只有两种
{
if(segtree[root].left == segtree[root].right)
{
segtree[root].maxn = val;
} //这也是终结的条件,到最底了。直接修改
else
{
if(x<= segtree[root<<1].right)
revise(root<<1,x,val); //如果小于左子树右边界进入左子树
else
revise(root<<1|1,x,val); //不然进入右子树
segtree[root].maxn=max(segtree[root<<1].maxn,segtree[root<<1|1].maxn);
}
return ;
}
int main()
{
int l,r,x,val,n,i,m;
char ops[20];
while(scanf("%d%d",&n,&m)!=EOF)
{
for(i=1;i<=n;i++)
{
scanf("%d",&a[i]);
}
CreateSegmentree(1,1,n);
while(m--)
{
scanf("%s",ops);
if(ops[0]=='Q')
{
scanf("%d%d",&l,&r);
printf("%d\n",query(1,l,r));
}
else
{
scanf("%d%d",&x,&val);
revise(1,x,val);
}
}
}
return 0;
}