HDU 5806 二分

NanoApe Loves Sequence Ⅱ

 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

题解：用前缀和保存前i个区间有多少不小于m的数，对于左端点for过去，然后二分又端点，如果二分到某个又端点恰好满足(l,r)之中第k大数不小于m，那么又端点右边的区间都符合。本题也可以用尺取法来做。

#include<iostream>
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
typedef long long ll;
int a[200005],n,m,k,s1[200005];
int check(ll l,ll r){
    if(s1[r]-s1[l-1]>=k)return true;
    else return false;
}
int main(){
    int t,i,j;
    scanf("%d",&t);
    while(t--){
        scanf("%d%d%d",&n,&m,&k);
        memset(s1,0,sizeof(s1));
        for(i=1;i<=n;i++){
            scanf("%d",&a[i]);
            s1[i]+=s1[i-1]+(a[i]>=m?1:0);
        }
        ll sum=0;
        for(i=1;i<=n-k+1;i++){
            ll l=i+k-1,r=n;
            while(l+1<r){
                ll mid=(l+r)/2;
                if(check(i,mid)){
                    r=mid;
                }
                else l=mid; 
            }
            if(check(i,l))sum+=n-l+1;
            else if(check(i,r))sum+=n-r+1;
        }
        printf("%lld\n",sum);
    }
    return 0;
}