首页 > 代码库 > CodeVS 1081 线段树练习 2

CodeVS 1081 线段树练习 2

2024-07-30 13:44:23   219人阅读

1081 线段树练习 2

 

 时间限制: 1 s
 空间限制: 128000 KB
 题目等级 : 大师 Master
 
 
 
题目描述 Description

给你N个数，有两种操作


1：给区间[a,b]的所有数都增加X


2：询问第i个数是什么？

输入描述 Input Description

第一行一个正整数n，接下来n行n个整数，再接下来一个正整数Q，表示操作的个数. 接下来Q行每行若干个整数。如果第一个数是1，后接3个正整数a,b,X，表示在区间[a,b]内每个数增加X,如果是2，后面跟1个整数i, 表示询问第i个位置的数是多少。

输出描述 Output Description

对于每个询问输出一行一个答案

样例输入 Sample Input

3

1

2

3

2

1 2 3 2

2 3

样例输出 Sample Output

5

数据范围及提示 Data Size & Hint

数据范围

1<=n<=100000

1<=q<=100000

 

解题：线段树的基本操作。。。

 1 #include <iostream> 2 #include <cstdio> 3 #include <cstring> 4 #include <cmath> 5 #include <algorithm> 6 #include <climits> 7 #include <vector> 8 #include <queue> 9 #include <cstdlib>10 #include <string>11 #include <set>12 #include <stack>13 #define LL long long14 #define pii pair<int,int>15 #define INF 0x3f3f3f3f16 using namespace std;17 const int maxn = 100010;18 struct node {19     int lt,rt,val;20 };21 node tree[maxn<<2];22 void build(int lt,int rt,int v) {23     tree[v].lt = lt;24     tree[v].rt = rt;25     if(lt == rt) {26         scanf("%d",&tree[v].val);27         return;28     }29     tree[v].val = 0;30     int mid = (lt + rt)>>1;31     build(lt,mid,v<<1);32     build(mid+1,rt,v<<1|1);33 }34 void update(int lt,int rt,int v,int val) {35     if(tree[v].lt >= lt && tree[v].rt <= rt) {36         tree[v].val += val;37         return;38     }39     if(tree[v].val) {40         tree[v<<1].val += tree[v].val;41         tree[v<<1|1].val += tree[v].val;42         tree[v].val = 0;43         return;44     }45     if(rt >= tree[v<<1|1].lt) update(lt,rt,v<<1|1,val);46     if(lt <= tree[v<<1].lt) update(lt,rt,v<<1,val);47 }48 int query(int lt,int rt,int v) {49     if(tree[v].lt >= lt && tree[v].rt <= rt) {50         return tree[v].val;51     }52     if(tree[v].val) {53         tree[v<<1].val += tree[v].val;54         tree[v<<1|1].val += tree[v].val;55         tree[v].val = 0;56     }57     if(lt <= tree[v<<1].rt) return query(lt,rt,v<<1);58     if(rt >= tree[v<<1|1].lt) return query(lt,rt,v<<1|1);59 }60 int main() {61     int n,m,op,a,b,x;62     while(~scanf("%d",&n)) {63         build(1,n,1);64         scanf("%d",&m);65         while(m--) {66             scanf("%d",&op);67             if(op == 1) {68                 scanf("%d %d %d",&a,&b,&x);69                 update(a,b,1,x);70             } else if(op == 2) {71                 scanf("%d",&a);72                 printf("%d\n",query(a,a,1));73             }74         }75     }76     return 0;77 }
View Code

 

CodeVS 1081 线段树练习 2


联系
我们