学习博客
知乎算法笔记
比较简单的一种数据结构,但是感觉非常好用
可以快速查询区间最值
如
f
[
i
]
[
j
]
表
示
以
a
[
i
]
开
始
长
度
为
2
j
的
数
里
边
的
最
大
值
,
即
f
[
i
]
[
j
]
=
m
a
x
(
a
[
i
]
,
.
.
.
,
a
[
i
+
2
j
]
)
如f[i][j]表示以a[i]开始长度为2^{j}的数里边的最大值,即f[i][j]=max(a[i],...,a[i+2^{j}])
如f[i][j]表示以a[i]开始长度为2j的数里边的最大值,即f[i][j]=max(a[i],...,a[i+2j])
合并操作:
f
[
i
]
[
j
]
=
m
a
x
(
f
[
i
]
[
j
−
1
]
,
f
[
i
+
(
1
<
<
(
j
−
1
)
)
]
[
j
−
1
]
)
f [ i ] [ j ] = max ( f [ i ] [ j - 1 ] , f [ i + ( 1 << ( j - 1 ) ) ] [ j - 1 ] )
f[i][j]=max(f[i][j−1],f[i+(1<<(j−1))][j−1])
查询操作:
int s = Log[r - l + 1];// 满足 2 * s >= r - l + 1
int Max = max(st[l][s], st[r - (1 << s) + 1][s];// 左端点=右端点-长度+1
// l为起点长度为2的s次方的一段
// r为终点,长度为2的s次方的一段
// 这两端大多数情况下都会交叉,极端情况下恰好覆盖 r - l + 1 这段,
也可以维护区间 g c d gcd gcd
King of Range
题意:
给定
n
n
n 个数,
m
m
m 个查询,每次查询给出一个
k
k
k,求有多少区间
[
l
,
r
]
[l,r]
[l,r] 满足
a
m
a
x
−
a
m
i
n
>
k
a_{max}-a_{min}>k
amax−amin>k,即区间极差大于
k
k
k
思路:
ST表+单调指针
枚举左端点,右端点是单调的
j
j
j 指针初始设为
1
1
1,
j
j
j 指针至多移动
n
n
n 次
code:
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn = 1e5 + 9;
ll n, m, k;
int Log[maxn], stmax[maxn][25], stmin[maxn][25];
void init()// 必须预处理,不然会t
{
Log[1] = 0;
for(int i = 2; i <= n; ++i)
Log[i] = Log[i >> 1] + 1;
}
bool find(int l, int r, ll k)// 判断区间极差是否大于k
{
int s = Log[r - l + 1];
int ma = max(stmax[l][s], stmax[r - (1 << s) + 1][s]);
int mi = min(stmin[l][s], stmin[r - (1 << s) + 1][s]);
if(ma - mi > k) return 1;
else return 0;
}
int main()
{
cin >> n >> m;
init();
for(int i = 1; i <= n; ++i)
{
scanf("%d", &stmax[i][0]);
stmin[i][0] = stmax[i][0];
}
for(int j = 1; j <= 21; ++j)// 预处理st表 枚举长度 2的j次方
for(int i = 1; i + (1 << j) - 1 <= n; ++i)
{ // 左端点+长度-1=右端点
stmax[i][j] = max(stmax[i][j - 1], stmax[i + (1 << (j - 1))][j - 1]);// i+长度=下一段的左端点,如果这里-1,就成了i为起点的这段的右端点
stmin[i][j] = min(stmin[i][j - 1], stmin[i + (1 << (j - 1))][j - 1]);
}
while(m--)
{
scanf("%lld", &k);
ll ans = 0;
// 重点思维:j指针是单调的,没必要每次从i开始
for(int i = 1, j = 1; i <= n; ++i)// 固定左端点不动,寻找最小的右端点
{
while(!find(i, j, k) && j <= n) ++j;
//cout << j << endl;
if(j <= n) ans += n - j + 1;// 最小的右端点,这时区间i到j的极差大于k,
else break;
}
printf("%lld\n", ans);
}
return 0;
}
单调队列+双指针
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int n,m,k;
int qmax[100100],qmin[100100];
int a[100100];
int main(){
cin >> n >> m;
for(int i=1;i<=n;i++) cin >> a[i];
for(int i = 1; i <= m; i++)
{
cin >> k;
ll ans = 0;
int hmax = 1, hmin = 1,tmax = 0,tmin = 0;
int l = 1;
for(int j = 1; j <= n; j++)
{
while(hmax <= tmax && a[j] >= a[qmax[tmax]]) tmax--;//单调递减,队头为最大值
while(hmin <= tmin && a[j] <= a[qmin[tmin]]) tmin--; //单调递增,队头为最小值
// 找到 a[j]可以插入的位置
qmax[++tmax] = j;// 插入 a[j]
qmin[++tmin] = j;// tmax和tmin 是队尾指针
//cout<<a[mmax[hmax]]<<" "<<a[mmin[hmin]]<<endl;
while(hmax <= tmax && hmin <= tmin && a[qmax[hmax]] - a[qmin[hmin]] > k)
{
ans += n - j + 1;// j是满足条件的离 l最近的位置
//左端点为l,扫到 j时满足最值差 > k ,左端点 l不变,加上 j以后的数,最值差不会减小,所以贡献值为 n - j + 1
l++;
if(qmax[hmax] < l) hmax++;
if(qmin[hmin] < l) hmin++;
// cout << " " << l << " " << ans << " " << endl;
}
}
cout << ans << endl;
}
}
D. Integers Have Friends
先求差分数组,然后预处理ST表维护区间gcd
枚举左端点,二分查找右端点维护最大子区间长度
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 2e5 + 9;
ll n, m, k;
ll f[maxn][25];
ll Log[maxn];
ll d[maxn], a[maxn];
void init()
{
for(int i = 2; i <= maxn - 5; ++i)
Log[i] = Log[i >> 1] + 1;
}
ll query(ll l, ll r)
{
ll s = Log[r - l + 1];
return (ll)__gcd(f[l][s], f[r - (1 << s) + 1][s]);
}
void work()
{
cin >> n;
for(int i = 1; i <= n; ++i)
{
scanf("%lld", &a[i]);
if(i > 1)
d[i-1] = abs(a[i] - a[i - 1]), f[i-1][0] = d[i-1];
}
--n;
for (int j = 1; j <= Log[n]; ++j)
for(int i = 1; i + (1 << j) - 1 <= n; ++i)
f[i][j] = __gcd(f[i][j - 1], f[i + (1 << (j - 1))][j - 1]);
ll s = 0;
for(ll i = 1; i <= n; ++i)
{
if(f[i][0] == 1) continue;
ll l = i, r = n;
while(l < r)// 这个写法是 二分查找最后一个满足条件的数的模板
{
ll m = (l + r + 1) >> 1;
if(query(i, m) == 1) r = m - 1;
else l = m;
}
s = max(s, l - i + 1);
}
cout << s + 1ll << endl;
}
int main()
{
init();
int T;cin >> T;while(T--)
work();
return 0;
}