[算法] 区间问题

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

1. 前缀和

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

  • 使用前缀和可在O(1)时间计算区间和:
    由通项公式b[i] = \sum_{k=1}^{i} a[k]可得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),即前缀和不支持单点修改和区间修改

二维/多维前缀和

前缀和基于容斥原理,下面以二维前缀和为例说明前缀和扩展到多维的方式。
例如一个二维数组 a
1 2 4 3
5 1 2 4
6 3 5 9
前缀和矩阵即 sum_{x,y} = \sum\limits_{i=1}^x \sum\limits_{j=1}^y a_{i,j}
1 3 7 10
6 9 15 22
12 18 29 45
二维前缀和的初始化即递推求 sum
sum_{i,j} = sum_{i - 1,j} + sum_{i,j - 1} - sum_{i - 1,j - 1} + a_{i,j}

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

sum_i 表示结点 i 到根节点的权值总和。
若是点权, x,y 路径上的和为 sum_x + sum_y - sum_{lca} - sum_{fa_{lca}}
若是边权, x,y 路径上的和为 sum_x + sum_y - 2sum_{lca}

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. 差分+前缀和
    满足互相看见约束的序列各位置的最高可能高度
树上差分

基于子树和而非节点到根的和
如果使 x,y 路径上的点权增加 k
b_x = b_x + k
b_y = b_y + k
b_{lca} = b_{lca} - k
b_{fa_{lca}} = b_{fa_{lca}} - k

如果是边权
b_x = b_x + k
b_y = b_y + k
b_{lca} = b_{lca} - 2k

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. 莫队算法

用于离线区间

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 204,445评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,889评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 151,047评论 0 337
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,760评论 1 276
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,745评论 5 367
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,638评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,011评论 3 398
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,669评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,923评论 1 299
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,655评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,740评论 1 330
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,406评论 4 320
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,995评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,961评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,197评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,023评论 2 350
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,483评论 2 342

推荐阅读更多精彩内容