树状数组(一维、二维、区间修改)(含封装结构体)

树状数组封装

单点修改,区间查询

struct Fenwick {
    const int len;
    vector<int> a;
    Fenwick(int n1) : len(n1), a(len + 1, 0) {}
    void init(vector<int>& b)  //O(n)初始化
    {
        for(int i = 1; i <= len; i++)
        {
            a[i] += b[i];
            int j = i + ((i) & (-i));
            if(j <= len)
                a[j] += a[i];
        }
    }
    void add(int x, int y) 
    {
        for(int i = x; i <= len; i += ((i) & (-i)))
            a[i] += y;
    }
    int sum(int x)
    {
        int ans = 0;
        for(int i = x; i; i -= ((i) & (-i)))
            ans += a[i];
        return ans;
    }
    int getsum(int l, int r)
    {
        return sum(r) - sum(l - 1); 
    }
};

区间修改,区间查询

struct Fenwick {
    const int len;
    vector<int> c1,c2;
    Fenwick(int n1,vector<int> &a) : len(n1), c1(n1+1,0), c2(n1+1,0) //vector a初始化a为1下标
    {
        for(int i=1;i<=n1;i++)
            add(i,a[i]-a[i-1]);
    }
    Fenwick(int n1) : len(n1), c1(n1+1,0), c2(n1+1,0){} //初始化0
    int lowbit(int x)
    {
        return ((x)&(-x));
    }
    void add(int x,int y)
    {
        for(int i=x;i<=n;i+=lowbit(i))
        {
            c1[i]+=y;
            c2[i]+=x*y;
        }
    }
    int getsum(int x)
    {
        int res=0;
        for(int i=x;i;i-=lowbit(i))
            res+=(x+1)*c1[i]-c2[i];
        return res;
    }
    void change(int l,int r,int d) //区间修改:[l,r]+d
    {
        add(l,d);
        add(r+1,-d);
    }
    int rangesum(int l,int r)//区间查询:[l,r]
    {
        return getsum(r)-getsum(l-1);
    }
};
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值