蓝书（算法竞赛进阶指南）刷题记录——POJ3468 A Simple Problem with Intergers（树状数组维护差分）

关注

原创最新推荐文章于 2026-03-21 00:47:27 发布 191 阅读 AI 写同款· GEO 优化 ›

基于MATLAB的电池等效电路模型（ECM）仿真，用于分析电气行为和动态响应.zip 1.版本：matlab2014a/2019b/2024b 2.附赠案例数据可直接运行。 3.代码特点：参数化编程、参数可方便更改、代码编程思路清晰、注释明细。 4.适用对象：计算机，电子信息工程、数学等专业的大学生课程设计、期末大作业和毕业设计。立即下载

题目：POJ3468.
题目大意：给定一个长度为 $n$ 的序列 $a$ ，要求支持：
1.格式 $C\,a\,b\,c$ ，表示将 $[a, b]$ 的权值都加上 $c$ .
2.格式 $Q\,a\,b$ ，表示查询 $[a, b]$ 的权值和.
设操作数为 $m$ ，那么 $1\leq n,m\leq 10^5$ .

线段树裸题，可是我们这里不用线段树，我们讨论树状数组的解法.

我们已经知道树状数组支持单点修改区间查询和区间修改单点查询，那么可不可以让树状数组支持区间修改区间查询呢？

我们现在继续将思路留在差分数组上，我们设 $A$ 是 $a$ 的差分数组，那么 $A [1, n]$ 的和就是：
$\sum_{i=1}^{n}\sum_{j=1}^{i}A[j]=\sum_{i=1}^{n}A[i]*(n-i+1).$

我们发现这个东西用一个树状数组还是不好维护，我们可以再变式：
$\sum_{i=1}^{n}A[i]*(n-i+1)=\sum_{i=1}^{n}A[i]*n-\sum_{i=1}^{n}A[i]*(i-1).$

我们发现这个东西貌似就可以用两个树状数组维护，我们可以用一个树状数组维护 $A [i]$ ，再用一个树状数组维护 $A [i] (i - 1)$ .

代码如下：

//#include<bits/stdc++.h>
#include<iostream>
  using namespace std;
#define Abigail inline void
typedef long long LL; 
const int N=100000;
LL c[N+9][2],a[N+9];
int n,q;
int lowbit(int x){return x&-x;}
void add(int x,LL num,int t){
  while (x<=n){
    c[x][t]+=num;
    x+=lowbit(x);
  }
}
LL query(int x,int t){
  LL sum=0;
  while (x){
    sum+=c[x][t];
    x-=lowbit(x);
  }
  return sum;
}
char rc(){
  char c=getchar();
  while (c<'A'||c>'Z') c=getchar();
  return c;
}
Abigail into(){
  scanf("%d%d",&n,&q);
  for (int i=1;i<=n;i++)
    scanf("%lld",&a[i]);
}
Abigail work(){
  for (int i=n;i>=1;i--)
    a[i]-=a[i-1];
  for (int i=1;i<=n;i++)
    add(i,a[i],0),add(i,a[i]*(i-1),1);
}
Abigail getans(){
  char opt;
  int l,r;
  LL d;
  for (int i=1;i<=q;i++){
    opt=rc();
    if (opt=='C'){
      scanf("%d%d%lld",&l,&r,&d);
      add(l,d,0);add(r+1,-d,0);
      add(l,d*(l-1),1);add(r+1,-d*r,1);
    }else{
      scanf("%d%d",&l,&r);
      printf("%lld\n",query(r,0)*r-query(r,1)-query(l-1,0)*(l-1)+query(l-1,1));
    }
  }
}
int main(){
  into();
  work();
  getans();
  return 0;
}