树状数组(可以修改的前缀和)

树状数组可以办到

  • 区间查询
  • 单点查询
  • 单点修改

就是做不到区间修改
但是呢,区间修改,你完全完全可以直接用树状数组的单点修改去做,就是时间复杂度会稍微慢了一丢丢。无所谓的。
但是!!这里有一个可以实现和线段树一样快的树状数组的方法!!

树状数组就是一个前缀和,树状数组和前缀和没有任何差别,就是当你需要对于一个数组维护,一边修改,修改完再查前缀和,需要这样的操作的时候你再去使用树状数组。这样会比较节省时间和空间,要是你只有看前缀和的话,你还是就直接用前缀和算了。

前缀和呢,其实也是完全可以看成是可以求一段区间很好求的东西。

其中有且只有3个很简单的函数

  • lowbit函数,求x的二进制表达式中最低位的的1所对应的值x&(-x)
  • add函数,参数有2个,x,v表示第x个数加上v
    循环,从这个数到最大是数N(这个要特别注意一下),里面就是正常的加哦 tr[i] += v
    注意,add函数的循环和query函数的循环每一步的幅度都是lowbit(i),注意这里lowbit(i)实战中背错一次!!!
  • sum函数,参数就1x,表示前x个数相加的和
    循环,i = xi(i>0)变小的循环,res += tr[i]虽然是前缀和,但是却不是+=a[i]哦,这里是树状数组领域!

注意点

  • 你需要两个数组,一个是a存的是你每一个的数字,一个是你的前缀和数组tr,因为名字叫树状数组,所以也就起一个tr这个好听的名字(tree)
  • 1.读入每一个数字2.add(i,a[i])这两步是连续进行的,注意,下标都是从1开始,这一点很重要,下标必须从1开始,如果他的输入是有可能从0开始的,那你就x++
  • 有一个巨重要巨重要的点,就是你输入a[i]的时候要同时进行add(i,a[i])操作!!!!千万不要忘记了!!!
  • 对于树状数组添加的维护,一开始就要进行!!!
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

const int N = 100010;

int n, m;
int a[N], tr[N];

int lowbit(int x)
{
    return x & -x;
}

void add(int x, int v)
{
    for (int i = x; i <= N; i += lowbit(i)) tr[i] += v;
}

int sum(int x)
{
    int res = 0;
    for (int i = x; i; i -= lowbit(i)) res += tr[i];
    return res;
}

int main()
{
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; i ++ ) scanf("%d", &a[i]);
    for (int i = 1; i <= n; i ++ ) add(i, a[i]);

    while (m -- )
    {
        int k, x, y;
        scanf("%d%d%d", &k, &x, &y);
        if (k == 0) printf("%d\n", sum(y) - sum(x - 1));
        else add(x, y);
    }

    return 0;
}

这,就是可以实现和线段树一样的快速区间修改的树状数组!


#include<bits/stdc++.h>
using namespace std;
long long n,m,a[200010],b[200010],c[200010],bb[200010],c1[200010];
long long lowbit(long long x)
{
	return x&-x;
}
void modify(int x,long long y)
{
	for (int i=x;i<=n;i+=lowbit(i))
		b[i]+=y;
}
long long query(long long x)
{
	long long ans=0;
	for (int i=x;i>0;i-=lowbit(i))
		ans+=b[i];
	return ans;
}
void modifyy(long x,long long y)
{
	for (int i=x;i<=n;i+=lowbit(i))
		bb[i]+=y;
}
long long queryy(long long x)
{
	long long ans=0;
	for (int i=x;i>0;i-=lowbit(i))
		ans+=bb[i];
	return ans;
}
int main()
{
	cin>>n>>m;
	for (int i=1;i<=n;i++)
	{
		cin>>a[i];
		c[i]=a[i]-a[i-1];
		c1[i]=(i-1)*c[i];
	}
	
	for (int i=1;i<=n;i++)
		modify(i,c[i]);
	for (int i=1;i<=n;i++)
		modifyy(i,c1[i]);
	
	for (int i=1;i<=m;i++)
	{
		long long p;
		cin>>p;
		if (p==1)
		{
			long long xx,yy,zz;
			cin>>xx>>yy>>zz;
			modify(xx,zz);
			modify(yy+1,-zz);
			modifyy(xx,(xx-1)*zz);
			modifyy(yy+1,-yy*zz);
		}
		if (p==2)
		{
			long long xx,yy,ss;
			cin>>xx>>yy;
			ss=(yy*query(yy)-queryy(yy))-((xx-1)*query(xx-1)-queryy(xx-1));
			cout<<ss<<endl;
		}
	}
	return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值