HDU 5806 NanoApe Loves Sequence Ⅱ

最新推荐文章于 2017-08-19 04:03:33 发布

原创最新推荐文章于 2017-08-19 04:03:33 发布 · 344 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#二分

本文介绍了一种筛选子字符串的有效算法，通过枚举左端点并使用二分查找确定右端点来找出所有符合条件的子串，即子串中第k大的数不小于给定值m。文中提供了一个具体实现示例，并对比了尺取法等其他解决方案。

题意，给你一个串问你该串有几个连续子串满足该子串的第k大的数不小于m

方法，统计每个位置（包括当前位置）前面有几个大于等于m的数，一个区间只要满足该区间大于 m 的数有k个及以上，就是合法的

我的做法是枚举左端点，二分寻找右端点，nlogn的效率，看了别人的，用了尺取法，只要n就可以了，确实不错，自己做的时候没想到

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

int num[200005];
int sum[200005];
int n, m, k;

int biSearch(int ll, int rr, int val)
{
	int l = ll, r = rr;
	int mid;
	while (l <= r)
	{
		mid = (l + r) >> 1;
		if (sum[mid] >= val) r = mid - 1;
		else l = mid + 1;
	}
	return l;
}

int main(void)
{
	int T;
	scanf("%d", &T);
	while (T--)
	{
		int counter = 0;
		long long ans = 0;
		scanf("%d%d%d", &n, &m, &k);
		for (int i = 0; i < n; i++)
			scanf("%d", &num[i]);
		for (int i = 0; i < n; i++) 
		{
			if (num[i] >= m)
				counter++;
			sum[i] = counter;
		}
		ans += n - biSearch(k - 1, n - 1, k);
		for (int i = 1; i < n - k + 1; i++)
		{
			int pos = biSearch(i + k - 1, n - 1, k + sum[i - 1]);
			ans += n - pos;
		}
		printf("%lld\n", ans);
	}
	return 0;
}