[算法] 区间问题

本文对区间查询问题常用的数据结构方法进行总结

1. 前缀和

前缀和是降低区间查询问题复杂度的一种常见预处理方法,对数组a进行前缀和初始化需要O(n)时间:新建数组b,将数组a的累加依次放入数组b中
b[0] = a[0]
b[i] = b[i-1] + a[i]

  • 使用前缀和可在O(1)时间计算区间和:
    由通项公式可得sum[i, j] = b[j] - b[i-1]
  • 但前缀和单点修改:
void modify(int pos, int d) {
  for (int i = pos; i <=n; ++i) {
    b[i] += d;
  }
}

时间复杂度为O(n),即前缀和不支持单点修改和区间修改

二维/多维前缀和

前缀和基于容斥原理,下面以二维前缀和为例说明前缀和扩展到多维的方式。
例如一个二维数组
1 2 4 3
5 1 2 4
6 3 5 9
前缀和矩阵即
1 3 7 10
6 9 15 22
12 18 29 45
二维前缀和的初始化即递推求

  • 使用二维前缀数组可在O(1)时间求左上(x1,y1) - 右下(x2,y2) 子矩阵的和:
    同理即
树上前缀和

设 表示结点 到根节点的权值总和。
若是点权, 路径上的和为 ;
若是边权, 路径上的和为 。

LCA 最近公共祖先

2. 差分

差分可看作前缀和的逆操作 对原数组a求差分数组b,则a也是b的前缀和数组。
b[1] = a[1]
b[i] = a[i] - a[i-1]

a[i] = b[1] + b[2] + ... + b[i]
=a[1] + a[2] - a[1] +...+ a[i] - a[i-1] = a[i]
差分数列不能提供O(1)时间的区间和查询,但可以提供O(1)时间的单点修改和区间修改, 例如对区间[l, r]内所有的数都加d只需

void add (int l, int r, int d) {
  b[l] += d;
  b[r+1] -= d;
}

则原数组 a[1]...a[l-1]间的数均无影响,a[l]...a[r]间的数均增加d,a[r+1]后的数又抵消不变。
适用于区间修改、单点查询类问题

  • 差分例题:
  1. 差分+贪心
    求区间内所有值通过区间加1/减1全部相等的最小变换次数
  2. 差分+前缀和
    满足互相看见约束的序列各位置的最高可能高度
树上差分

基于子树和而非节点到根的和
如果使 路径上的点权增加 ,



如果是边权


3. 倍增法 ST表

定义:倍增是指在状态空间很大线性递推无法满足要求时,成倍增长只递推状态空间中在2的整数次幂位置上的值,其他值由于任意整数可表示为若干2的次幂项的和可以由上述代表值拼加而成。
应用:

  1. 快速幂基于倍增和二进制划分的思想
  2. RMQ(Range Maximum/Minimum Query)问题,解决RMQ主要方法即ST表和线段树。ST表基于倍增思想,可以通过O(nlogn)时间预处理,以O(1)时间解决每次重复区间对结果无影响的区间查询问题,例如区间最大值最小值,但不支持修改操作
  1. LCA问题

4. 树状数组

树状数组是支持区间单点修改的前缀和。不同于前缀和,树状数组求原数组2的幂次长度的和,即前1个,前2个,前4个和前8个的和,但这样我们无法求出[3,3]区间和[5.8]区间的子区间的和,因此我们需要对[5,8]区间也按照2的幂的长度划分即[5], [5,6], [5,6,7,8],类似[3,3]我们仍无法求[7,7],因此补上{3}和{7},如下图所示:


由于所有的区间长度都是2的幂次,即1,10,100,1000
将上图所有区间从左至右按序排列,其区间长度的二进制表示为:
1,10,1, 100, 1, 10, 1, 1000
而图中区间标号对应的二进制表示为:
1,10,11,100,101,110,111,1000
可以发现区间长度即区间标号二进制表示从右往左出现第一个1以及这个1之后的那些0组成数的二进制对应的十进制的数,用lowbit函数将区间标号映射为区间长度:

int lowbit (int x) {
  return x & (-x);
}
  • 单点修改操作需要更新所有包含它的区间,时间复杂度O(log n):
void modify(int pos, int d) {
  while(pos <= n) {
    b[pos] += d;
    pos += lowbit(pos);
  } 
}
  • 区间和
// a[1]...a[x]
int presum(int x) {
  int ans = 0;
  while(x >= 1) {
    ans += b[x];
    x -= lowbit(x);
  }
}

// a[i]...a[j]
int sum(int i, int j)
  return presum(j) - presum(i-1);

扩展:

  1. 使用差分转换 树状数组也可以支持区间修改、单点查询区间修改、区间查询
  2. 二维树状数组


BIT例题:

  1. 求逆序对
  2. 计算右侧小于当前元素的个数
  3. 二分树状数组

5. 线段树

线段树能解决综合上述所有需求的区间问题。

6. 莫队算法

用于离线区间

你可能感兴趣的:([算法] 区间问题)