HDU - 5806 NanoApe Loves Sequence Ⅱ 想法题

时间:2022-07-20 22:11:22

http://acm.hdu.edu.cn/showproblem.php?pid=5806

题意:给你一个n元素序列,求第k大的数大于等于m的子序列的个数。

题解:题目要求很奇怪,很多头绪但写不出,选择跳过的题,简称想法题。

首先考虑区间的更新方法:区间左端l不动,右端r滑动,

滑到有k个数>=m时,此区间符合条件,并且发现右端点再往右滑到底,此条件一直符合(因为若加入的数小于“第K大的数”,则毫无影响。若不然,加入该数会产生一个新的第k大数,保证>=“第K大的数”>=m)

所以一找到这个数,ans直接加上n-r+1+1;(要+2,因为r又被更新+1了一下,)

然后更新一次左区间,如果离开区间的数<n,则毫无影响,该区间继续符合条件。若不然,则重新更新右区间,直到再次找到一个>=k的数。

这样就是基本线性了。   如果r滑倒头后还找不到,那么gg,跳出循环。

ac代码:

#define    _CRT_SECURE_NO_WARNINGS
#include<stdio.h>
#include<string.h>
#include<iostream>
using namespace std;
const int maxn = 2e5 + ;
const int mod= 1e9 + ;
int a[maxn], x[maxn];
int main(){
int n, m, k;
int t;
cin >> t;
while (t--) {
cin >> n >> m >> k; for (int i = ; i <= n; i++) {
scanf("%d", &a[i]);
}
long long ans = ;
int l = , r = , cnt = ;
while () {
while(r <= n&&cnt < k) {
if (a[r] >= m)cnt++;
r++;
}
if (cnt < k) break;
ans += n - r + ;
if (a[l] >= m) cnt--;
l++;
}
cout << ans << endl;
} }