首页 > 代码库 > hdu1754 I Hate It(线段树)

hdu1754 I Hate It(线段树)

2024-08-07 03:05:46   220人阅读

题目意思：

http://acm.hdu.edu.cn/showproblem.php?pid=1754


很多学校流行一种比较的习惯。老师们很喜欢询问，从某某到某某当中，分数最高的是多少。
这让很多学生很反感。
不管你喜不喜欢，现在需要你做的是，就是按照老师的要求，写一个程序，模拟老师的询问。当然，老师有时候需要更新某位同学的成绩。
 
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

题目分析：

直接上线段树模板即可，还是已理解为主，代码有注释。

AC代码：

#include<iostream>
#include<cstdio>
#include<map>
#include<cstring>
#include<string>
#include<algorithm>
#include<queue>
#include<vector>
#include<stack>
#include<cstdlib>
#include<cctype>
#include<cstring>
#include<cmath>
#define MAX 210005
using namespace std;
int a[MAX];
struct node{
    int left,right;
    int mx;//最大值
}tr[MAX*3];//注意扩大数组
void Build(int l,int r,int k){//建立树
    tr[k].left=l;
    tr[k].right=r;
    if(l==r){
        tr[k].mx=a[l];//已到达叶节点
        return ;//记得要return
    }
    int mid=(l+r)/2;
    Build(l,mid,2*k);//建立左子树
    Build(mid+1,r,2*k+1);//建立右子树
    tr[k].mx=max(tr[k*2].mx,tr[k*2+1].mx);//保存区间里的最大值
}
int Query(int l,int r,int k){//查询区间为l，r
    if(l==tr[k].left&&tr[k].right==r)//查询区间覆盖当前结点
        return tr[k].mx;

    int mid=(tr[k].left+tr[k].right)/2;
    if(r<=mid) return Query(l,r,k*2);//在左子树
    if(mid<l) return Query(l,r,k*2+1);//在右子树
    if(l<=mid&&mid<r){//一半在左子树，一半在右子树
        return max(Query(l,mid,2*k),Query(mid+1,r,2*k+1));//
    }
}
void Update(int id,int x,int k){//根据id找a[id].mx=x;
    if(tr[k].left==tr[k].right){
        tr[k].mx=x;
        return;//记住return
    }
    int mid=(tr[k].left+tr[k].right)/2;
    if(id<=mid){//在左子树
        Update(id,x,2*k);
    }
    if(id>mid) Update(id,x,2*k+1);//在右子树
    tr[k].mx=max(tr[2*k].mx,tr[2*k+1].mx);//更新所有父节点
}
int main()
{
    int n,m,id,k;
    while(~scanf("%d%d",&n,&m)){
        for(int i=1;i<=n;i++){
            scanf("%d",&a[i]);
        }
        Build(1,n,1);
        char s[5];
        for(int i=1;i<=m;i++){
            scanf("%s%d%d",s,&id,&k);
            if(s[0]=='U') Update(id,k,1);
            if(s[0]=='Q') printf("%d\n",Query(id,k,1));
        }
    }
	return 0;
}



hdu1754 I Hate It(线段树)


联系
我们