I Hate It (线段树)

来源:互联网 发布:知乎每周精选 订阅 编辑:程序博客网 时间:2024/06/05 15:41
很多学校流行一种比较的习惯。老师们很喜欢询问,从某某到某某当中,分数最高的是多少。
这让很多学生很反感。

不管你喜不喜欢,现在需要你做的是,就是按照老师的要求,写一个程序,模拟老师的询问。当然,老师有时候需要更新某位同学的成绩。

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 61 2 3 4 5Q 1 5U 3 6Q 3 4Q 4 5U 2 9Q 1 5
Sample Output
5659          
Hint
Huge input,the C function scanf() will work better than cin 


代码:

#include<stdio.h>#include<string.h>#include<algorithm>#define MAX 200005using namespace std;struct node{    int vel;}tree[MAX*4];//定义线段树int n,m;int arr[MAX];/*功能:新建线段树root:当前线段树的根结点下标arr: 用来构造线段树的数组left:树组起始下标right:数组结束下标*/void build(int root,int arr[],int left,int right){    if(left==right){//找到叶结点        tree[root].vel=arr[left];        return;    }    int mid=(left+right)/2;    build(root*2,arr,left,mid); //递归构造左子树    build(root*2+1,arr,mid+1,right); //递归构造右子树     //根据左右子树根节点的值,更新当前根节点的值    tree[root].vel=max(tree[root*2].vel,tree[root*2+1].vel);}/*功能:线段树的区间查询root:当前线段树的根节点下标[nleft, nright]: 当前节点所表示的区间[qletf, qright]: 此次查询的区间*/int query(int root,int nleft,int nright,int qleft,int qright){    if(qleft<=nleft&&qright>=nright) //当前区间在查询区间内        return tree[root].vel;    else if(qright<nleft||qleft>nright) return 0;//查询区间于当前区间没有交集    int mid=(nleft+nright)/2;    //查询左子树和有子树最大值 去二者的最大值    return max(query(root*2,nleft,mid,qleft,qright),query(root*2+1,mid+1,nright,qleft,qright));}/*功能:更新线段树中某个叶子节点的值root:当前线段树的根节点下标[nleft, nright]: 当前节点所表示的区间pos: 待更新节点在原始数组arr中的下标num: 更新的值(原来的值换成num)*/void updateOne(int root,int nleft,int nright,int pos,int num){    if(nleft==nright){//找到相应的结点        tree[root].vel=num;//更新结点的值        return;    }    int mid=(nleft+nright)/2;    if(pos<=mid) updateOne(root*2,nleft,mid,pos,num); //在左子树中更新    else updateOne(root*2+1,mid+1,nright,pos,num); //在右子树中更新     //根据左右子树根节点的值,更新当前根节点的值    tree[root].vel=max(tree[root*2].vel,tree[root*2+1].vel);}int main(){    while(scanf("%d%d",&n,&m)!=EOF){        memset(arr,0,sizeof(arr));        memset(tree,0,sizeof(tree));        int i;        for(i=1;i<=n;i++)            scanf("%d",&arr[i]);        build(1,arr,1,n);        while(m--){            char s[5];            scanf("%s",s);            if(s[0]=='Q'){                int a,b;                scanf("%d%d",&a,&b);                int sum=query(1,1,n,a,b);                printf("%d\n",sum);            }            else {                int pos,num;                scanf("%d%d",&pos,&num);                updateOne(1,1,n,pos,num);            }        }    }return 0;}




原创粉丝点击