数列问题 HDU 5806 （尺取法）

最新推荐文章于 2020-02-07 21:35:31 发布

转载最新推荐文章于 2020-02-07 21:35:31 发布 · 699 阅读

数论同时被 2 个专栏收录

5 篇文章

订阅专栏

算法入门

1 篇文章

订阅专栏

本文介绍了一个尺取法的应用实例，通过给定的一组数值和参数m、k，采用尺取法求解有多少个连续子区间使得第k大的数不小于m。尺取法是一种解决特定类型区间问题的有效算法。

#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<k && ed+1<=n)
				cnt+=(a[++ed]>=m);//满足条件为1
			//若有k个大于m，加起来 
			if(cnt==k) 
				ans=ans+n-ed+1;
			// 前面都已经够数了，后面怎么添都可以
			//每走一步，如果最开头的数 大于m，减掉，继续往后找 
			cnt-=(a[st++]>=m); 
		}
		printf("%I64d\n",ans); 
	}
	return 0;
}