首页 > 代码库 > 线段树——hdu1754I Hate It

线段树——hdu1754I Hate It

2024-11-16 14:44:02   204人阅读

一、题目回顾

题目链接：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<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn = 800005;
int arr[maxn>>1];
struct SegTreeNode{
    int val;
};
SegTreeNode segTree[maxn];

int Max(int x,int y){return x>=y ? x:y;}
 
void build(int root,int arr[],int istart,int iend)
{
    if(istart == iend)                      //叶子结点
        segTree[root].val = arr[istart];    /* 只有一个元素,节点记录该单元素 */
    else{
        int mid = (istart + iend)/2;
        build(2*root,arr,istart,mid);       //递归构造左子树
        build(2*root+1,arr,mid+1,iend);     //递归构造右子树
        //根据左右子树根节点的值，更新当前根节点的值
        segTree[root].val = Max(segTree[2*root].val,segTree[2*root+1].val); /* 回溯时得到当前node节点的线段信息 */
    }
}

int query(int root, int nstart, int nend, int qstart, int qend)
{
    //查询区间和当前节点区间没有交集
    if(qstart > nend || qend < nstart)
        return -1;
    //当前节点区间包含在查询区间内
    if(qstart <= nstart && qend >= nend)
        return segTree[root].val;
    //分别从左右子树查询，返回两者查询结果的较小值
    int mid = (nstart + nend) / 2;
    return Max(query(root*2, nstart, mid, qstart, qend),
               query(root*2+1, mid + 1, nend, qstart, qend));
 
}
void updateOne(int root, int nstart, int nend, int index, int addVal)
{
    if(nstart == nend)
    {
        if(index == nstart)//找到了相应的节点，更新之
            segTree[root].val = addVal;
        return;
    }
    int mid = (nstart + nend) / 2;
    if(index <= mid)//在左子树中更新
        updateOne(root*2, nstart, mid, index, addVal);
    else updateOne(root*2+1, mid+1, nend, index, addVal);//在右子树中更新
    //根据左右子树的值回溯更新当前节点的值
    segTree[root].val = Max(segTree[root*2].val, segTree[root*2+1].val);
}

int main()
{
	int N,M;
	while(scanf("%d%d",&N,&M)!=EOF){
		for(int i=1;i<=N;i++){
			scanf("%d",&arr[i]);
		}
		getchar();	//6666666666666666666666666666666666666TL
		build(1,arr,1,N); 
		char ch;int a,b;
		while(M--){
			scanf("%c%d%d",&ch,&a,&b);
			getchar();				//6666666666666666666666666666666666
			if(ch==‘Q‘){
				printf("%d\n",query(1,1,N,a,b));
			}
			if(ch==‘U‘){
				updateOne(1,1,N,a,b);
			}
		}
	}
	return 0;
}

 

三、我的收获

①当输入与字符有关时，应考虑回车是否要处理；

②数组空间开的大小与TLE无关，只与MEL有关，也可能与RE有关。

线段树——hdu1754I Hate It


联系
我们