码迷,mamicode.com
首页 > 编程语言 > 详细

算法训练 操作格子 线段树板子题

时间:2017-04-02 22:44:04      阅读:267      评论:0      收藏:0      [点我收藏+]

标签:iostream   names   query   struct   ret   scan   using   区间   roo   

问题描述

有n个格子,从左到右放成一排,编号为1-n。

共有m次操作,有3种操作类型:

1.修改一个格子的权值,

2.求连续一段格子权值和,

3.求连续一段格子的最大值。

对于每个2、3操作输出你所求出的结果。

输入格式

第一行2个整数n,m。

接下来一行n个整数表示n个格子的初始权值。

接下来m行,每行3个整数p,x,y,p表示操作类型,p=1时表示修改格子x的权值为y,p=2时表示求区间[x,y]内格子权值和,p=3时表示求区间[x,y]内格子最大的权值。

输出格式

有若干行,行数等于p=2或3的操作总数。

每行1个整数,对应了每个p=2或3操作的结果。

样例输入
4 3
1 2 3 4
2 1 3
1 4 3
3 1 4
样例输出
6
3
数据规模与约定

对于20%的数据n <= 100,m <= 200。

对于50%的数据n <= 5000,m <= 5000。

对于100%的数据1 <= n <= 100000,m <= 100000,0 <= 格子权值 <= 10000。

 
  1 #include<bits/stdc++.h>
  2 #include<iostream>
  3 #include<cstring>
  4 #include<cstdio>
  5 #include<algorithm>
  6 #include<vector>
  7 #define ll __int64
  8 #define PI acos(-1.0)
  9 #define mod 1000000007
 10 using namespace std;
 11 int n,m;
 12 struct node
 13 {
 14     int l,r;
 15     int value;
 16     int maxn;
 17 }tree[400005];
 18 int exm;
 19 void  buildtree(int root ,int l,int r)
 20 {
 21      tree[root].l=l;
 22      tree[root].r=r;
 23     if(l==r)
 24     {
 25         scanf("%d",&exm);
 26         tree[root].value=tree[root].maxn=exm;
 27         return;
 28     }
 29     int mid=(l+r)>>1;
 30     buildtree(root<<1,l,mid);
 31     buildtree(root<<1|1,mid+1,r);
 32     tree[root].maxn=max(tree[root<<1].maxn,tree[root<<1|1].maxn);
 33     tree[root].value=tree[root<<1].value+tree[root<<1|1].value;
 34 }
 35 void update(int root ,int what,int c)
 36 {
 37     if(tree[root].l==tree[root].r&&tree[root].r==what)
 38     {
 39         tree[root].maxn=tree[root].value=c;
 40         return;
 41     }
 42     int mid=(tree[root].l+tree[root].r)>>1;
 43     if(what<=mid)
 44         update(root<<1,what,c);
 45     else
 46         update(root<<1|1,what,c);
 47     tree[root].maxn=max(tree[root<<1].maxn,tree[root<<1|1].maxn);
 48     tree[root].value=tree[root<<1].value+tree[root<<1|1].value;
 49 }
 50 int query1(int root,int l,int r)
 51 {
 52     if(tree[root].l==l&&tree[root].r==r)
 53     {
 54         return tree[root].value;
 55     }
 56     int mid=(tree[root].l+tree[root].r)>>1;
 57     if(r<=mid)
 58          return query1(root<<1,l,r);
 59     else
 60     {
 61         if(l>mid)
 62           return query1(root<<1|1,l,r);
 63         else
 64           return query1(root<<1,l,mid)+query1(root<<1|1,mid+1,r);
 65     }
 66 }
 67 int query2(int root,int l,int r)
 68 {
 69     if(tree[root].l==l&&tree[root].r==r)
 70     {
 71         return tree[root].maxn;
 72     }
 73     int mid=(tree[root].l+tree[root].r)>>1;
 74     if(r<=mid)
 75          return query2(root<<1,l,r);
 76     else
 77     {
 78         if(l>mid)
 79           return query2(root<<1|1,l,r);
 80         else
 81           return max(query2(root<<1,l,mid),query2(root<<1|1,mid+1,r));
 82     }
 83 }
 84 int main()
 85 {
 86     int q,w,e;
 87     scanf("%d %d",&n,&m);
 88     buildtree(1,1,n);
 89     for(int i=1;i<=m;i++)
 90     {
 91         scanf("%d %d %d",&q,&w,&e);
 92         if(q==1)
 93             update(1,w,e);
 94         if(q==2)
 95             printf("%d\n",query1(1,w,e));
 96         if(q==3)
 97             printf("%d\n",query2(1,w,e));
 98     }
 99     return 0;
100 }

 

算法训练 操作格子 线段树板子题

标签:iostream   names   query   struct   ret   scan   using   区间   roo   

原文地址:http://www.cnblogs.com/hsd-/p/6659981.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!