题意:求有多少个区间满足题目要求
思路一:
以i为起点的区间满足题意的有多少个
st+二分
#include <map>
#include <set>
#include <stack>
#include <queue>
#include <cmath>
#include <ctime>
#include <vector>
#include <cstdio>
#include <cctype>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
using namespace std;
#define INF 0x3f3f3f3f
#define inf -0x3f3f3f3f
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define mem0(a) memset(a,0,sizeof(a))
#define mem1(a) memset(a,-1,sizeof(a))
#define mem(a, b) memset(a, b, sizeof(a))
typedef long long ll;
int n;
int a[100100];
int d1[100100][31];
int d2[100100][31];
void RMQ_init(){
for(int i=1;i<=n;i++){
d1[i][0]=a[i];
d2[i][0]=a[i];
}
for(int j=1;j<=30;j++)
for(int i=0;i+(1<<j)-1<=n;i++){
d1[i][j]=min(d1[i][j-1],d1[i+(1<<(j-1))][j-1]);
d2[i][j]=max(d2[i][j-1],d2[i+(1<<(j-1))][j-1]);
}
}
int RMQ_min(int L,int R){
int k=log2(R-L+1);
return min(d1[L][k],d1[R-(1<<k)+1][k]);
}
int RMQ_max(int L,int R){
int k=log2(R-L+1);
return max(d2[L][k],d2[R-(1<<k)+1][k]);
}
int main(){
int t,k;
scanf("%d",&t);
while(t--){
scanf("%d%d",&n,&k);
for(int i=1;i<=n;i++){
scanf("%d",&a[i]);
}
RMQ_init();
ll ans=0;
for(int i=1;i<=n;i++){
int r=n,l=i;
while(l+1<r){
int p=(l+r)>>1;
if(RMQ_max(i,p)-RMQ_min(i,p)>=k)
r=p;
else
l=p;
}
if(RMQ_max(i,r)-RMQ_min(i,r)>=k)
ans+=ll(l-i+1);
else
ans+=ll(r-i+1);
}
printf("%I64d\n",ans);
}
return 0;
}
思路二:以i为结尾的区间满足题意的有多少个
#include <map>
#include <set>
#include <stack>
#include <queue>
#include <cmath>
#include <ctime>
#include <vector>
#include <cstdio>
#include <cctype>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
using namespace std;
#define INF 0x3f3f3f3f
#define inf -0x3f3f3f3f
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define mem0(a) memset(a,0,sizeof(a))
#define mem1(a) memset(a,-1,sizeof(a))
#define mem(a, b) memset(a, b, sizeof(a))
typedef long long ll;
int n;
int a[100100];
int d1[100100][31];
int d2[100100][31];
void RMQ_init(){
for(int i=1;i<=n;i++){
d1[i][0]=a[i];
d2[i][0]=a[i];
}
for(int j=1;j<=30;j++)
for(int i=0;i+(1<<j)-1<=n;i++){
d1[i][j]=min(d1[i][j-1],d1[i+(1<<(j-1))][j-1]);
d2[i][j]=max(d2[i][j-1],d2[i+(1<<(j-1))][j-1]);
}
}
int RMQ_min(int L,int R){
int k=log2(R-L+1);
return min(d1[L][k],d1[R-(1<<k)+1][k]);
}
int RMQ_max(int L,int R){
int k=log2(R-L+1);
return max(d2[L][k],d2[R-(1<<k)+1][k]);
}
int main(){
int t,k;
scanf("%d",&t);
while(t--){
scanf("%d%d",&n,&k);
for(int i=1;i<=n;i++){
scanf("%d",&a[i]);
}
RMQ_init();
int l=1;
ll ans=0;
for(int r=1;r<=n;r++){
while(RMQ_max(l,r)-RMQ_min(l,r)>=k) //枚举右端点
l++;
ans+=(r-l+1);
}
printf("%I64d\n",ans);
}
return 0;
}