POJ 3468 A Simple Problem with Integers (树状和差分,分块,线段树延迟标记,详细的解答)

题目意思:
Q:求L到R的和。C:L,R的数都加x

树状数组版:

看了网上很多代码,综合以下的解题步骤:

本题用到的知识点:树状数组,差分
简单介绍差分:
假设原数组为 a[1] ~ a[n]
用数组b[1] ~ b[n] 作为 a数组的差分数组
b[1] = a[1]; //第一项特殊些
b[2] = a[2] - a[1];

b[n] = a[n] - a[n - 1];

那么b的前缀和(加入前k项),刚好为数组a的第k项的值,a[k]
sum{b[1], b[2], … , b[k]} = b[1] + b[2] + … + b[k] = a[k] //简单代入即可证明

如果在数组a区间 [L, R] 的范围内,每个位置加上值x, 对于差分数组b,发生变化的只有两项,
b[L] += x, b[R + 1] -= x // b[k] = a[k] - a[k - 1]; 2 <= k <= n, 取遍每一个k,很容易得出结论

思路:
1、使用差分分解:
现在计算a数组的前k项 的和,
sum{a[1], a[2], …, a[k]} = a[1] + a[2] + … + a[k] //依次展开每个 a[i] (1 <= i <= k)
= (b[1]) + (b[1] + b[2]) + (b[1] + b[2] + b[3]) + … + (b[1] + b[2] + … + b[k]) //有k个小括号
= k * (b[1] + b[2] + … + b[k]) - 0 * b[1] - 1 * b[2] - 2 * b[3] - … - (k - 1) * b[k]
//每个括号加上一些数, 都变成 (b[1] + b[2] + … + b[k]), 然后减去那些数(一共加上了 0 项的 b[1], 1 项的 b[2], … , 依次类推)
= k * sum{b[1], b[2], … , b[k]} - sum{0 * b[1], 1 * b[2], 2 * b[3], … , (k - 1) * b[k]}
//把数组a的前k 项和 sum{a[1], a[2], …, a[k]} 分解为两部分了
// 第一部分 : 差分数组b的前k项之和 sum{b[1], b[2], … , b[k]} 的 k倍
// 第二部分 : 序列 (i - 1) * b[i] 的前k项的和 sum{0 * b[1], 1 * b[2], 2 * b[3], … , (k - 1) * b[k]}

2、 使用树状数组分析:
第一部分的和 sum{b[1], b[2], … , b[k]} ( 简化记为:sum(c[1], k) ),用 树状数组 c[1] 来记录;
第二部分的和sum{0 * b[1], 1 * b[2], 2 * b[3], … , (k - 1) * b[k]}(简化记为:sum(c[0], k)), 用 树状数组 c[0] 来记录

sum{a[1], a[2], ..., a[k]} = k * sum(c[1], k) - sum(c[0], k)

那么原始数组a, [L, R]区间上同时加上x就可以表示为:(差分数组的结论)
对于c[1]来说,在L位置上加上x,在R + 1位置上加上−x; 
对于c[0]来说,在L位置上加上−x * (L − 1),在R+1位置上加上x * R; 	
//用 i = L 和 i = R + 1 代入式子 (i - 1) * b[i]得到,因为是负号,取相反数

3、 查询结果时候,求 前k项的和公式为
long long sumk = k * getSum(k, 1) + getSum(k, 0);

4、 最后,使用long long

#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;
#define lowbit(i) ((i)&(-i))	// lowbit(x) 表示x的二进制对应的值
const int MaxN = 100010;
long long c[2][MaxN];			//树状数组c[1] 和 c[0]
int n, q;
int A[MaxN];

//getSum()函数, 返回前x个数的和
long long getSum(int x, int index)
{
   
	long long sum = 0;
	for(int i = x; i > 0; i-= lowbit(i))
	{
   
		sum += c[index][i];
	}
	return sum;
}

//update()函数,将第x个数加上v
void update(int x, int v, int index)
{
   
	for(int i = x; i < MaxN; i += lowbit(i))
	{
   
		c[index][i] += v;
	}
}

int main()
{
   
	char cmd[2] = {
   0};
	scanf("%d%d", &n, &q);
	memset(c, 0, sizeof(c));
	for(int i = 1; i <= n; ++i)
	{
   
		scanf("%d", &A[i]);	
		update(i, A[i], 0);	// 数组 c[0] 进行初始化
	}
	int s, e, x;
	for(int i = 0; i < q; ++i)
	{
   
		scanf("%s", cmd);	
		if(cmd[0] == 'Q')
		{
   
			scanf("%d%d", &s, &e);
			long long suml = (s - 1) * getSum(s - 1, 1) + getSum(s - 1, 0)
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值