首页 > 编程语言 > 详细

luogu_【模板】树状数组 2

时间:2019-11-02 17:07:05      阅读:61      评论:0      收藏:0      [点我收藏+]

树状数组记录修改,差分

#include<iostream>
#include<cstdio>

#define ri register int
#define u int

namespace opt {

    inline u in() {
        u x(0),f(1);
        char s(getchar());
        while(s<0||s>9) {
            if(s==-) f=-1;
            s=getchar();
        }
        while(s>=0&&s<=9) {
            x=(x<<1)+(x<<3)+s-0;
            s=getchar();
        }
        return x*f;
    }

}

using opt::in;

#define NN 500005

namespace mainstay {

    u N,K;

    u c[NN],a[NN];

    inline u ask(const u &x) {
        u _re(0);
        for(ri i(x); i; i-=i&-i) _re+=c[i];
        return _re;
    }

    inline void add(const u &x,const u &y) {
        for(ri i(x); i<=N; i+=i&-i) c[i]+=y;
    }

    inline void solve() {
        N=in(),K=in();
        for(ri i(1); i<=N; ++i) a[i]=in();
        for(ri i(1); i<=K; ++i) {
            u _k(in());
            if(_k==1) {
                u _a(in()),_b(in()),_c(in());
                add(_a,_c),add(_b+1,-_c);
            } else {
                u _a(in());
                printf("%d\n",ask(_a)+a[_a]);
            }
        }
    }

}

int main() {

    //freopen("x.txt","r",stdin);
    std::ios::sync_with_stdio(false);
    mainstay::solve();

}

 

luogu_【模板】树状数组 2

原文:https://www.cnblogs.com/ling-zhi/p/11782526.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!