树状数组整理

树状数组基本性质:
(1)C[i] = A[i - 2^k+1] + A[i - 2^k+2] + ... + A[i]; //k为i的二进制中从最低位到高位连续零的长度
(2)SUMi = C[i] + C[i-2^k1] + C[(i - 2^k1) - 2^k2] + .....;
(3)A[i] 包含于 C[i + 2^k]、C[(i + 2^k) + 2^k]...;

树状数组:a[maxn]是原序列,c[maxn]是树状数组的维护序列(存储一颗树)

基本运用
lowbit

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

无区间修改时
一、单点修改

单点修改
void updata(int i,long long k) //在i位置加上k
{   
    while(i <= n)
	{
        c[i] += k;
        i += lowbit(i);
    }
}

二、区间查询

区间查询
long long getsum(int i)////求A[1 到 i]的和
{        
    long long res = 0;
    while(i > 0)
    {
        res += c[i];
        i -= lowbit(i);
    }
    return res;
}
//l到r的区间和=getsum(r)-getsum(l-1);

三、单点查询:
value[k]=getsum(k)-getsum(k-1)
把一个点当作长度为1的区间即可

有区间修改时
一、区间修改&单点查询
直接修改每个值一定会复杂度爆炸
很容易想到,要用差分建树:由于在某一区间内加上或减去某一个值,区间内各个值之间的差是不变的。
因此只要修改区间头尾l和r+1两个点的值。
这样写复杂度就ok了。

实现:
规定A[0]=0;
则有 A[i] = ΣD[j](j从1一直增加到i);(D[j] = A[j] - A[j-1]),即前面i项的差值和。
当某个区间[x,y]值改变了,区间内的差值是不变的,只有D[x]和D[y+1]的值发生改变。
因此只修改x和y+1的正确性是显然的。

代码中选择对数组D建立树状数组

int n,m;
int a[50005] = {0},c[50005]; //对应原数组和树状数组

int lowbit(int x)
{
	return x&(-x);
}
 
void updata(int i,int k)
{    //在i位置加上k
	while(i <= n)
	{
		c[i] += k;
        i += lowbit(i);
    }
} 
int getsum(int i)
{        //求D[1 - i]的和,即A[i]值
	int res = 0;
	while(i > 0)
	{
		res += c[i];
		i -= lowbit(i);
    }     
	return res;
}
int main()
{
	cin>>n;
	for(int i = 1; i <= n; i++)
	{
        cin>>a[i];
        updata(i,a[i] - a[i-1]);   //输入初值的时候,也相当于更新了值
    }
    //[x,y]区间内加上k
    updata(x,k);    //A[x] - A[x-1]增加k
    updata(y+1,-k);        //A[y+1] - A[y]减少k
    //查询i位置的值
    int sum = getsum(i);
    return 0;
}

二、区间查询&&区间修改
既然有区间修改那么就一定要利用差分
已知如下
image

则A[1]+A[2]+...+A[n]

= (D[1]) + (D[1]+D[2]) + ... + (D[1]+D[2]+...+D[n])

= nD[1] + (n-1)D[2] +... +D[n]

= n * (D[1]+D[2]+...+D[n]) - (0D[1]+1D[2]+...+(n-1)*D[n])

因此最开始的公式可变为
image
维护两个数状数组,sum1[i] = D[i],sum2[i] = D[i]*(i-1);

int n,m;
int a[50005] = {0};
int sum1[50005];    //(D[1] + D[2] + ... + D[n])
int sum2[50005];    //(1*D[1] + 2*D[2] + ... + n*D[n])

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

void updata(int i,int k){
    int x = i;    //因为x不变,所以得先保存i值
    while(i <= n){
        sum1[i] += k;
        sum2[i] += k * (x-1);
        i += lowbit(i);
    }
}

int getsum(int i){        //求前缀和
    int res = 0, x = i;
    while(i > 0){
        res += x * sum1[i] - sum2[i];
        i -= lowbit(i);
    }
    return res;
}

int main(){
    cin>>n;
    for(int i = 1; i <= n; i++){
        cin>>a[i];
        updata(i,a[i] - a[i-1]);   //输入初值的时候,也相当于更新了值
    }

    //[x,y]区间内加上k
    updata(x,k);    //A[x] - A[x-1]增加k
    updata(y+1,-k);        //A[y+1] - A[y]减少k

    //求[x,y]区间和
    int sum = getsum(y) - getsum(x-1);

    return 0;
}
posted @ 2021-08-26 13:56  Mint-hexagram  阅读(60)  评论(0编辑  收藏  举报