信奥一本通-树状数组模版题目-修改数列元素+求子数列元素和

Posted ofshk

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了信奥一本通-树状数组模版题目-修改数列元素+求子数列元素和相关的知识,希望对你有一定的参考价值。

给定n个数列,规定有两种操作,一是修改某个元素,二是求子数列[a,b]的连续和。数列的元素个数最多10万个,询问操作最多10万次。

 
输入
第一行2个整数n,m(n表示输入n个数,m表示有m个操作)
第二行输入n个数列。
接下来m行,每行有三个数k,a,b(k=0表示求子数列[a,b]的和;k=1表示第a个数加b)。

 
输出
输出若干行数字,表示k=0时,对应的子数列[a,b]的连续和。


样例输入
10 5
1 2 3 4 5 6 7 8 9 10
1 1 5
0 1 3
0 4 8
1 7 5
0 4 8
 

样例输出
11
30
35
 

 

模板题。信奥p211

记住三步操作:求lowbit、对某个元素进行加法操作、查询某个前缀和。

 1 #include<stdio.h>
 2 #include<string.h>
 3 #include<algorithm>
 4 #include<cmath>
 5 #include<iostream>
 6 using namespace std;
 7 const int N=1e5+10;
 8 int n,c[N];
 9 
10 int lowbit(int x)
11 {
12     return x&(-x);
13 }
14 
15 void update(int x,int k)//在第x个位置(数)上加上k,并在树状数组中修改相应元素//对某个元素进行加法操作
16 {
17     while(x<=n)
18     {
19         c[x]+=k;
20         x+=lowbit(x);
21     }
22 }
23 
24 int sum(int x)//查询前缀和
25 {
26     int res=0;
27     while(x>0)
28     {
29         res+=c[x];
30         x-=lowbit(x);
31     }
32     return res;
33 }
34 
35 int main()
36 {
37     int m,k;
38     scanf("%d %d",&n,&m);
39     for(int i=1; i<=n; i++)
40     {
41         scanf("%d",&k);
42         update(i,k);
43     }
44     for(int i=1; i<=m; i++)
45     {
46         int kk,a,b;
47         scanf("%d %d %d",&kk,&a,&b);
48         if(kk==0)//求区间和且输出
49         {
50             int ans=sum(b)-sum(a-1);
51             printf("%d
",ans);
52         }
53         else if(kk==1)//把第a个位置的元素更改为b
54             update(a,b);
55     }
56     return 0;
57 }

 

以上是关于信奥一本通-树状数组模版题目-修改数列元素+求子数列元素和的主要内容,如果未能解决你的问题,请参考以下文章

一本通.1536数星星 Stars(树状数组)

树状数组基本模版(区间更新,单点查询)

树状数组模板题:一本通1535

《信奥一本通》提高版—简要题解

#10118 「一本通 4.1 练习 3」打鼹鼠

POJ 3468(树状数组+区间修改)