【HDU5806 BestCoder Round 86C】【双指针】多少个区间内第k大数不小于m

NanoApe Loves Sequence Ⅱ

 

 Accepts: 374

 

 Submissions: 946

 Time Limit: 4000/2000 MS (Java/Others)

 

 Memory Limit: 262144/131072 K (Java/Others)

问题描述

退役狗 NanoApe 滚回去学文化课啦！

在数学课上，NanoApe 心痒痒又玩起了数列。他在纸上随便写了一个长度为 $n$ 的数列，他又根据心情写下了一个数 $m$。

他想知道这个数列中有多少个区间里的第 $k$ 大的数不小于 $m$，当然首先这个区间必须至少要有 $k$ 个数啦。

输入描述

第一行为一个正整数 $T$，表示数据组数。

每组数据的第一行为三个整数 $n,m,k$。

第二行为 $n$ 个整数 A_iA​i​​，表示这个数列。

1 \le T \le 10,~2 \le n \le 200000,~1 \le k \le n/2,~1 \le m,A_i \le 10^91≤T≤10, 2≤n≤200000, 1≤k≤n/2, 1≤m,A​i​​≤10​9​​

输出描述

对于每组数据输出一行一个数表示答案。

输入样例

1
7 4 2
4 2 7 7 6 5 1

输出样例

18

#include<stdio.h>
#include<iostream>
#include<string.h>
#include<string>
#include<ctype.h>
#include<math.h>
#include<set>
#include<map>
#include<vector>
#include<queue>
#include<bitset>
#include<algorithm>
#include<time.h>
using namespace std;
void fre() { freopen("c://test//input.in", "r", stdin); freopen("c://test//output.out", "w", stdout); }
#define MS(x,y) memset(x,y,sizeof(x))
#define MC(x,y) memcpy(x,y,sizeof(x))
#define MP(x,y) make_pair(x,y)
#define ls o<<1
#define rs o<<1|1
typedef long long LL;
typedef unsigned long long UL;
typedef unsigned int UI;
template <class T1, class T2>inline void gmax(T1 &a, T2 b) { if (b>a)a = b; }
template <class T1, class T2>inline void gmin(T1 &a, T2 b) { if (b<a)a = b; }
const int N = 2e5 + 10, M = 0, Z = 1e9 + 7, ms63 = 0x3f3f3f3f;
template <class T1, class T2>inline void gadd(T1 &a, T2 b) { a = (a + b) % Z; }
int casenum, casei;
int n, m, k;
int a[N], sum[N];
int main()
{
	scanf("%d", &casenum);
	for (casei = 1; casei <= casenum; ++casei)
	{
		scanf("%d%d%d", &n, &m, &k);
		for (int i = 1; i <= n; ++i)
		{
			scanf("%d", &a[i]);
			a[i] = (a[i] >= m);
			sum[i] = sum[i - 1] + a[i];
		}
		LL ans = 0;
		int r = 1;
		for (int l = 1; l <= n; ++l)
		{
			while (r <= n&&sum[r] - sum[l - 1] < k)++r;
			if (r > n)break;
			ans += (n - r + 1);
		}
		printf("%lld\n", ans);
	}
	return 0;
}
/*
【题意】
问你有多少个区间的第k大的数>=m

【类型】
双指针

【分析】
>=m的数记为1，然后用双指针求多少个区间内的数之和>=k

【时间复杂度&&优化】
O(n)

*/