数列问题 HDU 5806 (尺取法)

#include "cstdio"
#include "iostream"
#include "cstring"
#include "algorithm"
using namespace std;
/*题意:
给一个数列,给定一组m,k值,求有多少个人区间里第k大的数不小于m 
做法:
尺取法 
*/
int a[200010];
int main()
{
	int T;scanf("%d",&T);
	while(T--)
	{
		int n,m,k;
		scanf("%d%d%d",&n,&m,&k);
		//第k大的数,不小于m 
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&a[i]);
		}
		int st=1,ed=0,cnt=0;
		long long ans=0;
		while(st<=n)//从第一个遍历到最后 
		{
			//一旦够数,就停止加 
			while(cnt=m);//满足条件为1
			//若有k个大于m,加起来 
			if(cnt==k) 
				ans=ans+n-ed+1;
			// 前面都已经够数了,后面怎么添都可以
			//每走一步,如果最开头的数 大于m,减掉,继续往后找 
			cnt-=(a[st++]>=m); 
		}
		printf("%I64d\n",ans); 
	}
	return 0;
}

你可能感兴趣的:(算法入门,数论)