[洛谷2068] 統計和

題目描述

給定一個長度爲n（n<=100000），初始值都爲0的序列，x(x<=10000)次的修改某些位置上的數字，每次加上一個數，而後提出y (y<=10000)個問題，求每段區間的和。時間限制1秒。

輸入輸出格式

輸入格式：

第一行1個數，表示序列的長度n

第二行1個數，表示操做的次數w

後面依次是w行，分別表示加入和詢問操做

其中，加入用x表示，詢問用y表示

x的格式爲"x a b" 表示在序列a的位置加上b

y的格式爲"y a b" 表示詢問a到b區間的加和

輸出格式：

每行一個數，分別是每次詢問的結果

輸入輸出樣例

輸入樣例#1： 複製

5
4
x 3 8
y 1 3
x 4 9
y 3 4

輸出樣例#1： 複製

8
17

思路

線段樹；

代碼實現

 1 #include<cstdio>
 2 const int maxn=1e5+10;
 3 int n,m;
 4 int t[maxn];
 5 char ch;
 6 int x,y,z;
 7 void add(int k,int l,int r){
 8     if(l==r){t[k]+=y;return;}
 9     int mid=l+r>>1,ls=k<<1,rs=ls|1;
10     if(x<=mid) add(ls,l,mid);
11     else add(rs,mid+1,r);
12     t[k]=t[ls]+t[rs];
13 }
14 void tot(int k,int l,int r){
15     if(x<=l&&r<=y){z+=t[k];return;}
16     int mid=l+r>>1,ls=k<<1,rs=ls|1;
17     if(x<=mid) tot(ls,l,mid);
18     if(y>mid) tot(rs,mid+1,r);
19 }
20 int main(){
21     scanf("%d%d",&n,&m);
22     while(m--){
23         while(ch=getchar(),ch!='x'&&ch!='y');
24         if(ch=='x'){
25             scanf("%d%d",&x,&y);
26             add(1,1,n);
27         }
28         if(ch=='y'){
29             scanf("%d%d",&x,&y);
30             z=0,tot(1,1,n);
31             printf("%d\n",z);
32         }
33     }
34     return 0;
35 }