I Hate It 【线段树】

来源:互联网 发布:大连美工培训班哪里好 编辑:程序博客网 时间:2024/05/29 19:09

I Hate It
很多学校流行一种比较的习惯。老师们很喜欢询问,从某某到某某当中,分数最高的是多少。
这让很多学生很反感。

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

Hint
Huge input,the C function scanf() will work better than cin
思路: 单源点更新,求最大值
代码

#include<stdio.h>#include<string.h>#include<algorithm>#include<iostream>#include<math.h>#include<queue>#include<stack>#include<map>#include<vector>#include<set>#define LL long long#define M  5000000   //  定义的大 #define inf 0x3f3f3f3f #define mod 100009#define ll o<<1 #define rr o<<1|1   #define lson o<<1,l,mid    //  左子节点 #define rson o<<1|1,mid+1,r  //  右子节点   (易错写) using namespace std;struct Tree{    int l,r; //  一个节点  所代表的 区间左值和区间右值    int max;   // 一个节点 所代表的区间 的 权值 这里是 最大值 } tree[M<<2];  //   结构体 数组 的大小  注意 void pushup(int o){    tree[o].max = max(tree[ll].max,tree[rr].max);   // 更新 非子叶 节点  } void build(int o,int l, int r)   //构建  树   {    tree[o].l=l;  //      获取 当前节点的 区间左值     tree[o].r=r;   //     获取  当前节点的 区间右值     if( l== r)   //  如果 区间左值==区间右值 则 说明到了 子叶节点  可以从键盘获取  权值     {        int a;scanf("%d",&a);        tree[o].max=a;        return ;   //  不能少  ,判出的标志     }    int mid=( l+ r)>>1;    build(lson);  // 构建 左节点     build(rson);  //  构建 右节点     pushup(o);   //  每个非 子叶 节点都会进行更新  }  void update(int o,int pos,int val)   // 更新操作  {    if(tree[o].l==tree[o].r)     {        tree[o].max=val;        return ;   //  判出的标志  勿忘      }     int mid=(tree[o].l+tree[o].r)>>1;     if(pos<=mid) update(ll,pos,val); // 判定 要更新值的 位置 到底是在 左节点还是  右节点      else update(rr,pos,val);     pushup(o);  //  每个非 子叶 节点都会进行更新  } int query(int o,int l,int r)   //查 询操作 ===》 是从根节点 开始 寻找的  {    if(l<=tree[o].l&&r>=tree[o].r)  // 完全包含 才可以 判出     return tree[o].max;    int mid=(tree[o].l+tree[o].r)>>1;      //   由于 要查询的 区间不同  可能导致的 三种不同的 方向     if(r<=mid) query(ll,l,r);   // 可以判定 pos  一定在当前节点的 左子树     else if(l>mid ) query(rr,l,r);  //  可以判定 pos  一定在当前节点 的右子树     else     return max(query(ll,l,mid),query(rr,mid+1,r)); //  区间是在 左子树和右子树的中间,要找到其中的最大值  }int main(){    int n,m;    while(~scanf("%d%d",&n,&m))    {        build(1,1,n); // 定义 树 的起始          while(m--)         {            char c[2];int x,y;              scanf("%s%d%d",c,&x,&y);            if(c[0]=='Q') printf("%d\n",query(1,x,y));            else             update(1,x,y);         }    }    return 0;}

不用结构体的线段树,代码

#include<bits/stdc++.h>using namespace std;#define  LL long long#define fread() freopen("in.txt","r",stdin)#define fwrite() freopen("out.txt","w",stdout)#define CLOSE() ios_base::sync_with_stdio(false)const int MAXN = 200000+10;const int MAXM = 1e6;const int mod = 1e9+7;const int inf = 0x3f3f3f3f;int Max[MAXN<<2];void Up(int o){    Max[o]=max(Max[o<<1],Max[o<<1|1]);}void Build(int o,int le,int ri){    if(le==ri){        scanf("%d",&Max[o]);        return ;    }     int mid=(le+ri)>>1;    Build(o<<1,le,mid);    Build(o<<1|1,mid+1,ri);    Up(o) ; }void UpDate(int o,int le,int ri,int p,int val){    if(le==ri){        Max[o]=val;        return ;    }    int mid=(le+ri)>>1;    if(p<=mid) UpDate(o<<1,le,mid,p,val);    if(p>mid)  UpDate(o<<1|1,mid+1,ri,p,val);    Up(o); }int  Query(int o,int le,int ri,int L,int R){    if(L<=le&&ri<=R){        return Max[o];    }    int mid=(le+ri)>>1;    int ans=0;    if(L<=mid) ans=max(ans,Query(o<<1,le,mid,L,R));    if(R>mid) ans=max(ans,Query(o<<1|1,mid+1,ri,L,R));    return ans;}int main(){    CLOSE();//  fread();//  fwrite();    int n,m;    while(scanf("%d%d",&n,&m)!=EOF){        Build(1,1,n);        char op[2];int a,b;        while(m--){            scanf("%s %d %d",op,&a,&b);            if(op[0]=='Q')                 printf("%d\n",Query(1,1,n,a,b));            else UpDate(1,1,n,a,b);         }    }    return 0;}
0 0
原创粉丝点击