线段树线段树

1.I Hate It

2019-07-25  本文已影响0人  miaozasnone
  1. I Hate It

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

输入

本题目包含多组测试,请处理到文件结束。
在每个测试的第一行,有两个正整数 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。

输出

对于每一次询问操作,在一行里面输出最高成绩。

样例输入

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

样例输出

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

ac代码

#include<iostream>
#include<cstdio>
#include<cmath>
#define INF 0x7fffffff
#define maxn 4000001
using namespace std;
int arr[maxn];
int tree[maxn];
int arr_size,node_size;
void build_tree(int start,int end,int node){
    if(start==end){
        tree[node]=arr[start];
    }
    else{
        int mid=(start+end)/2;
        int left_node=node<<1;
        int right_node=node<<1|1;

        build_tree(start,mid,left_node);
        build_tree(mid+1,end,right_node);
        tree[node]=max(tree[left_node],tree[right_node]);

    }

}
void update_tree(int start,int end,int node,int val,int index){
    if(start==end){
        arr[index]=val;
        tree[node]=val;
        //cout<<node<<"--"<<tree[node]<<endl;
        return;
    }
    else{
        int mid=(start+end)/2;
        int left_node=node<<1;
        int right_node=node<<1|1;
        if(index>=start&&index<=mid){
            update_tree(start,mid,left_node,val,index);
        }
        else{
            update_tree(mid+1,end,right_node,val,index);
        }
        tree[node]=max(tree[right_node],tree[left_node]);
        // cout<<left_node<<"--"<<tree[left_node]<<" ";
        // cout<<right_node<<"--"<<tree[right_node]<<endl;
        // cout<<node<<"--"<<tree[node]<<endl;
    }
}
int querry_tree(int start,int end,int node,int l,int r){
    //  cout<<"start: "<<start<<"-->"<<end<<" "<<tree[node]<<endl;
    //  cout<<"range: "<<l<<"-->"<<r<<endl;
    if(r<start||end<l)return 0;
    if(start==end)return tree[node];
    if(l<=start&&r>=end)return tree[node];
    else
    {
        int mid=(start+end)/2;
        int left_node=node<<1;
        int right_node=node<<1|1;
        int sum_left=querry_tree(start,mid,left_node, l, r);
        int sum_right=querry_tree(mid+1,end,right_node,l,r);
        return max(sum_left,sum_right);
    }
    return 0;
}
int main(){
    int n,m;
    while (cin>>n>>m)
    {
        for(int i=1;i<=n;i++){
            scanf("%d",&arr[i]);
        }
        build_tree(1,n,1);
        //for(int i=1;i<2*n-1;i++)cout<<tree[i]<<" ";
        for(int i=0;i<m;i++){
            char c;
            int k1,k2;
            getchar();
            scanf("%c%d%d",&c,&k1,&k2);
            if(c=='Q'){
                cout<<querry_tree(1,n,1,k1,k2)<<endl;
            }
            else{
                update_tree(1,n,1,k2,k1);
                // for(int i=0;i<2<<(int)log2(n)+1;i++)cout<<tree[i]<<" ";
                // cout<<endl;
                // for(int i=0;i<n;i++)cout<<arr[i]<<" ";
                // cout<<endl;
            }
        }
    }
}
上一篇 下一篇

猜你喜欢

热点阅读