欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

hdu1754(线段树入门题)

程序员文章站 2022-03-14 23:10:21
...

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

题目大意十分明了;
解题思路:单点修改,区间查询求最值,当然就是用线段树来维护
当然单纯的套用模板可不行,需要修改几个地方,
1.这个题不需要lazy标签,所以pushdown函数就不需要了(就是下传lazy标签的函数)
2.普通的线段树 tree[]数组里面存的是sum,这里就存max
3.pushup函数(更新当前节点的值)也需要修改,变为tree[rt]=max(tree[rt<<1],tree[rt<<1|1]);,还有查询函数也稍微修改


代码:

#include <bits/stdc++.h>

using namespace std;
const int maxn=2e5+5;
int n,m;
int tree[maxn<<2];
int a[maxn];
void pushUp(int rt,int l,int r)
{
    tree[rt]=max(tree[rt<<1],tree[rt<<1|1]);
}
void build(int rt,int l,int r)
{
    if(l==r)
    {
        tree[rt]=a[l];
        return;
    }
    int mid=(l+r)>>1;
    build(rt<<1,l,mid);
    build(rt<<1|1,mid+1,r);
    pushUp(rt,l,r);
}
int query(int rt,int l,int r,int L,int R)
{
    if(L<=l && R>=r)
    {
        return tree[rt];
    }
    int m1,m2,mid;
    m1=m2=0;
    mid=(l+r)>>1;
    if(L<=mid) m1=query(rt<<1,l,mid,L,R);
    if(R>mid) m2=query(rt<<1|1,mid+1,r,L,R);
    return max(m1,m2);
}
void update(int rt,int l, int r,int x,int y)
{
    if(l==r && l==x)//到达根节点,并且根节点为x
    {
        tree[rt]=y;
        return;
    }
    int mid=(l+r)>>1;
    if(x>=l && x<=mid)
        update(rt<<1,l,mid,x,y);//往左边查询
    else
        update(rt<<1|1,mid+1,r,x,y);//往右边查询
    pushUp(rt,l,r);//更新自己
}
int main()
{
    while(scanf("%d %d",&n,&m)==2)
    {
        memset(tree,0,sizeof tree);
        for(int i=1;i<=n;i++)
            scanf("%d",&a[i]);
//            记得是在输入后再build树
        build(1,1,n);
        char c;
        int x,y;
        while(m--)
        {
            getchar();
            scanf("%c %d %d",&c,&x,&y);
            if(c=='Q')
                printf("%d\n",query(1,1,n,x,y));
            else
                update(1,1,n,x,y);
        }
    }
    return 0;
}

相关标签: 算法竞赛