题意:
给你n个数字,求有多少个子区间满足
1.区间最大值 - 区间长度 ≤ k
2.区间内各元素不相同
题解:
今天学了启发式分治,这道题就可以用这个算法
1.每次找到该区间的最大值的位置,以此作为分割点,枚举区间短的部分作为端点,统计答案,然后两个区间再分治,再统计两个区间的答案
2.找区间最大值,我们可以直接用ST表,然后还要预处理出以每个点作前缀和后缀不同元素区间最长长度
具体看代码实现
AC_code:
#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e6+5;
int a[maxn], pre[maxn], suf[maxn], vis[maxn];
int st[maxn][21], lg[maxn];
long long ans;
int n, k;
int cnt = 0;
int getmax(int l, int r) {
int h = lg[r-l+1];
if(a[st[l][h]] >= a[st[r-(1<<h)+1][h]]) {
return st[l][h];
} else {
return st[r-(1<<h)+1][h];
}
}
void init() {//预处理出每个值是2的几次方
lg[0] = -1;
for(int i = 1; i < maxn; i++) {
lg[i] = lg[i>>1] + 1;
}
}
void ST() {//ST表预处理
for(int i = 1; i <= n; i++) {
st[i][0] = i;
}
for(int i = 1; i <= 20; i++) {
for(int j = 1; j <= n+1-(1<<i); j++) {
if(a[st[j][i-1]] >= a[st[j+(1<<(i-1))][i-1]]) {
st[j][i] = st[j][i-1];
} else {
st[j][i] = st[j+(1<<(i-1))][i-1];
}
}
}
}
void pre_suf() {//预处理前缀不同数的最大长度 后缀不同数的最大长度
for(int i = 1; i <= n; i++) {
vis[i] = 0;
}
pre[1] = 1;
vis[a[1]] = 1;
for(int i = 2; i <= n; i++) {
if(vis[a[i]]) {
pre[i] = max(pre[i-1], vis[a[i]]+1);
} else {
pre[i] = pre[i-1];
}
vis[a[i]] = i;
}
for(int i = 1; i <= n; i++) {
vis[i] = 0;
}
suf[n] = n;
vis[a[n]] = n;
for(int i = n-1; i >= 1; i--) {
if(vis[a[i]]) {
suf[i] = min(suf[i+1], vis[a[i]]-1);
} else {
suf[i] = suf[i+1];
}
vis[a[i]] = i;
}
}
void merge(int l, int r) {
if(l > r) {
return ;
}
int pos = getmax(l, r);//以a[pos]为最大值
// printf("%d %d------\n", l, r);
// printf("%d\n", pos);
if(pos - l < r - pos) { //枚举区间小的
for(int i = l; i <= pos; i++) { //以i为左端点
int t = a[pos] - k;
int rr = i + t - 1;
int R = min(r, suf[i]); //右端点最大可以取的值
rr = max(rr, pos);//右端点最小可以取的值
if(rr > R) {
continue;
}
ans += R - rr + 1; //右端点可以取的值的长度
}
} else {
for(int i = pos; i <= r; i++) {//以i为右端点
int t = a[pos] - k;
int ll = i - t + 1;
int L = max(l, pre[i]); //右端点最大可以取的值
ll = min(ll, pos);//右端点最小可以取的值
if(ll < L) {
continue;
}
ans += ll - L + 1; //右端点可以取的值的长度
}
}
merge(l, pos-1);
merge(pos+1, r);
}
int main() {
int t;
scanf("%d", &t);
init();
while(t--) {
scanf("%d%d", &n, &k);
ans = 0;
for(int i = 1; i <= n; i++) {
scanf("%d", &a[i]);
}
ST();
pre_suf();
merge(1, n);
printf("%lld\n", ans);
}
return 0;
}