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

树状数组2 - 区间加 单点求和

时间:2019-02-17 22:03:11      阅读:174      评论:0      收藏:0      [点我收藏+]

标签:long   std   amp   数组   names   false   class   分数   ++i   

树状数组 = O(logn) 单点修改 ,O(logn) 区间查询

如果要做到 区间修改 单点查询 我们就要加入差分的思想

用树状数组记录数组的差分 然后对差分进行前缀和就可以得到单点的数据

//ios::sync_with_stdio(false); 
#include<bits/stdc++.h> 
#define ll long long
using namespace std;
const int MAXN = 500010;
int n,m;
ll C[MAXN]; 
ll lowbit(ll x)
{
    return x&-x;
}
void add(ll i,ll val)
{
    while(i<=n){
        C[i]+=val;
        i +=lowbit(i);
    }
}
ll sum(ll i)
{
    ll s = 0;
    while(i>0){
        s+=C[i];
        i-=lowbit(i);
    }
    return s;
} 

int main(){
    cin >> n >> m;
    ll l,r,op,k;
    ll now = 0;
    for(int i=1;i<=n;++i){
        cin >> k;
        add(i,k-now);
        now = k;
    }//初始化记录差分数组
    for(int i=0;i<m;++i){
        cin >> op ;
        if(op==1){
            cin >> l >> r >> k;
            add(l,k);
            add(r+1,-k); //区间修改
        }
        else{
            cin >> k;
            cout << sum(k) <<endl;  //单点查询
        }
    }
    return 0;
}

树状数组2 - 区间加 单点求和

标签:long   std   amp   数组   names   false   class   分数   ++i   

原文地址:https://www.cnblogs.com/xxrlz/p/10392823.html

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