HDU 1754 I Hate It（线段树）

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

思路：线段树比树状数组思路上要清晰，但是编程复杂度要来得高。

代码：

#include <stdio.h>
#include <string.h>
#define max(a,b) ((a)>(b)?(a):(b))
const int N = 200005;

int n, m, num;
struct Tree {
int l, r, value;
}st[3 * N];

void build(int l, int r, int id) {
if (l != r) {
int mid = (l + r) / 2;
build(l, mid, 2 * id);
build(mid + 1, r, 2 * id + 1);
}
st[id].l = l; st[id].r = r; st[id].value = 0;
}

void Insert(int num, int value, int id) {
if (st[id].l != st[id].r) {
int mid = (st[id].l + st[id].r) / 2;
if (num <= mid) Insert(num, value, 2 * id);
else Insert(num, value, 2 * id + 1);
st[id].value = max(st[2 * id].value, st[2 * id + 1].value);
}
else
st[id].value = value;
}

int Query(int l, int r, int id) {
if (l == st[id].l && r == st[id].r)
return st[id].value;
int mid = (st[id].l + st[id].r) / 2;
if (l <= mid && r > mid) {
int a = Query(l, mid, 2 * id);
int b = Query(mid + 1, r, 2 * id + 1);
return max(a, b);
}
else if (l <= mid && r <= mid) {return Query(l, r, 2 * id);}
else {return Query(l, r, 2 * id + 1);}
}

void init() {
build(1, n, 1);
for (int i = 1; i <= n; i++) {
scanf("%d", &num);
Insert(i, num, 1);
}
}

void solve() {
char s[10];
int a, b;
while (m--) {
scanf("%s%d%d", s, &a, &b);
if (s[0] == 'Q') printf("%d\n", Query(a, b, 1));
else Insert(a, b, 1);
}
}

int main() {
while (~scanf("%d%d", &n, &m)) {
init();
solve();
}
return 0;
}