HDU-1754 I Hate It

I Hate It

Time Limit: 9000/3000 MS (Java/Others)
Memory Limit: 32768/32768 K (Java/Others)

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.

Reference Code

#include<cstdio>
#include<algorithm>
#define max std::max
const int maxn=2e5+3;
struct node{
	int l,r,v;
}tree[maxn*4];
void build(int rt,int l,int r){
	tree[rt].l=l;
	tree[rt].r=r;
	if (l==r){
		scanf("%d",&tree[rt].v);
		return;
	}
	int mid=(l+r)>>1;
	build(rt<<1,l,mid);
	build(rt<<1|1,mid+1,r);
	tree[rt].v=max(tree[rt<<1].v,tree[rt<<1|1].v);
}
void update(int rt,int a,int b){
	int l=tree[rt].l,r=tree[rt].r;
	if (l==r){
		tree[rt].v=b;
		return;
	}
	int mid=(l+r)>>1;
	if (mid>=a) update(rt<<1,a,b);
    else update(rt<<1|1,a,b);
	tree[rt].v=max(tree[rt<<1].v,tree[rt<<1|1].v);
}
int query(int rt,int a,int b){
	int l=tree[rt].l,r=tree[rt].r;
	if (l==a&&r==b)
		return tree[rt].v;
	int mid=(l+r)>>1;
	if (mid>=b) return query(rt<<1,a,b);
	else if(mid<a) return query(rt<<1|1,a,b);
	else return max(query(rt<<1,a,mid),
                    query(rt<<1|1,mid+1,b));
}
int main(){
	int n,m;
	while (~scanf("%d%d",&n,&m)){
		build(1,1,n);
		char Q;
		int a,b;
		while (m--){
			scanf(" %c%d%d",&Q,&a,&b);
			if (Q=='U') update(1,a,b);
			else printf("%d\n",query(1,a,b));
		}
	}
	return 0;
}

Tips

这是一条线段树的裸题，写这个主要是想总结一下线段树的知识。

首先来说，线段树主要解决区间问题，比如求给定区间的和、最大值、最小值等等。本质上，线段树是一棵二叉树，其基本操作包括建立、更新、查询。以下是存储区间$1$到$8$的线段树的结构图：

其中每一个节点都是一个结构体，结构体包括三个值l(left),r(right),v(value)，上图只标出了l和r。
线段树的写法多种多样，网上的模板也是五花八门，但其本质都是二分和递归。所以只要心中有结构图，代码应该就不是问题。
另外，在开辟线段树数组的时候需要注意，由于线段树是二分地存储区间信息的，因此需要开辟四倍的空间。

不难发现，其实结构体数组的下标和结构体的l和r是有一定关系的，因此也可以直接用一个int数组来实现线段树，只需在原有基础上给函数加两个参数（l和r）即可。