【算法模板】数据结构:ST表
概念
ST
表是一种基于 倍增 思想,用于解决 可重复贡献问题 的数据结构
可重复贡献问题 是指对于运算 optopt,满足 𝑥opt𝑥=𝑥x optx =x ,则对应的区间询问就是一个可重复贡献问题。例如,最大值有 max(𝑥,𝑥)=𝑥max(x ,x )=x ,gcd 有 gcd(𝑥,𝑥)=𝑥gcd(x ,x )=x,所以 RMQ 和区间 GCD 就是一个可重复贡献问题。像区间和就不具有这个性质,如果求区间和的时候采用的预处理区间重叠了,则会导致重叠部分被计算两次,这是我们所不愿意看到的。另外,optopt 还必须满足结合律才能使用 ST 表求解。
ST 表基于倍增 思想,可以做到 𝑂(n logn) 预处理,O*(1) 回答每个询问。但是不支持修改操作。
所谓预处理,即我们需要预先处理出,对于每个位置 i,我们需要知道,以 i 作为左端点, i+2^j-1作为右端点的每个区间的最大值。
由于 log2n 较小,所以我们可以直接枚举每个 j。
我们用一个数组表示,f[i][j]
表示以 i 为左端点,区间长度为2^j ,也就是右端点是 i+2^j-1。
模板
cpp
vector<vector<int>> ST_prework(const vector<int> &v) {
int n = v.size();
int t = log2(n)+1;
vector<vector<int>> ST(n, vector<int>(t));
for (int i = 0; i < n; i++)ST[i][0] = v[i];
for (int j = 1; j < t; j++)//枚举区间长度
for (int i = 0; i + (1 << j) - 1 < n; i++)
ST[i][j] = max(ST[i][j - 1], ST[i + (1 << (j - 1))][j - 1]);
return ST;
}
int ST_query(vector<vector<int>> &ST,int l, int r) {
int k = log2(r - l + 1);//左闭右闭
return max(ST[l][k], ST[r - (1 << k) + 1][k]);
}
例题
区间最大值
cpp
#include <bits/stdc++.h>
using namespace std;
int main() {
int n,q;cin>>n>>q;
const int t=log2(n)+1;
vector<vector<int>> st(n,vector<int>(t));
for(int i=0;i<n;i++)cin>>st[i][0];
for(int i=1;i<t;i++)
for(int j=0;j+(1<<(i-1))<n;j++)
st[j][i]=max(st[j][i-1],st[j+(1<<(i-1))][i-1]);
while(q--){
int l,r;cin>>l>>r;
l--,r--;
int x=log2(r-l+1);
cout<<max(st[l][x],st[r-(1<<x)+1][x])<<endl;
}
return 0;
}
Balanced Lineup G
USACO07JAN Balanced Lineup G - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)
cpp
#include <bits/stdc++.h>
using namespace std;
int main() {
int n, q;
cin >> n >> q;
const int t = log2(n) + 1;
vector<vector<int>> stmax(n, vector<int>(t));
for (int i = 0; i < n; i++)cin >> stmax[i][0];
auto stmin = stmax;
for (int j = 1; j < t; j++) {
for (int i = 0; i + (1 << (j - 1)) < n; i++) {
stmax[i][j] = max(stmax[i][j - 1], stmax[i + (1 << (j - 1))][j - 1]);
stmin[i][j] = min(stmin[i][j - 1], stmin[i + (1 << (j - 1))][j - 1]);
}
}
while(q--){
int a,b;cin>>a>>b;
--a,--b;
int x=log2(b-a+1);
int ans=0;
ans+=max(stmax[a][x],stmax[b-(1<<x)+1][x]);
ans-=min(stmin[a][x],stmin[b-(1<<x)+1][x]);
cout<<ans<<endl;
}
return 0;
}