挑选子串#蓝桥杯

1621、挑选子串

#include<bits/stdc++.h>
using namespace std;
/*
双指针适用于快速查找，匹配，排序或移动操作(并非用指针实现，是用两个变量表示下标操作)
(1)对撞指针
：left和right指针同时移动一个向右，一个向左，相遇时跳出循环
(2)快慢指针
：快指针r，慢指针l，构成区间[l,r]
*/
const int N =2e5+9;
int a[N];
void solve(const int &Case){
	int n,m,k;cin>>n>>m>>k;
	for(int i=1;i<=n;i++)cin>>a[i];
	int ans=0;
	//快慢指针
	for(int i=1,j=0,cnt=0;i<=n;i++){
		//i>j 说明区间不合法,j+1 表示向右移还有空间,cnt<k 表示不满足>=k
		while(i>j||(j+1<=n&&cnt<k)){
			cnt+=(a[++j]>=m);
		}
		if(cnt>=k){
			//满足条件的情况下,找到这个最小区间[i,j],有n-j+1个子串
			ans+=n-j+1;
		}
		//a[i]>=m的话,就重新向后找到满足的cnt
		cnt-=(a[i]>=m);
	}
	cout<<ans<<"\n";
}
int main(){
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	cout.tie(nullptr);
	int T=1;
	for(int Case=1;Case<=T;Case++)solve(Case);
	return 0;
}