二指针_____NanoApe Loves Sequence II(hdu 5806 BestCoder Round #86)

问题描述

退役狗 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

分析：其实要使得一个区间的第k大的数大于等于m只需要让这个区间大于等于m的数至少有k个,那么读入数据的时候让大于等于m的数为1，小于m的数为0。只需要求数列中区间和大于等于K的区间个数即可。那么我们可以使用二指针。

代码：

#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <iostream>
#include <string>
#include <math.h>
#include <vector>
#include <queue>
#include <map>
using namespace std;
int a[200010];
int main()
{
    int t,n,m,k,step;
    cin >> t;
    while(t--)
    {
        cin >> n >> m >> k;
        for(int i = 0 ; i < n ; i ++)
        {
            scanf("%d",&step);
            if(step >= m) a[i] = 1;
            else a[i] = 0;
        }
        int l,r,num;
        long long ans = 0;
        for(l = 0 ,r = 0,num = 0; l < n ; l ++)
        {
            while(r < n && num < k)num+=a[r++];
            if(num < k)break;
            ans = ans + n - r + 1;
            num -= a[l];
        }
        printf("%lld\n",ans);
    }
    return 0;
}