k小数查询(牛客练习赛87-B，思维)

k小数查询

思路：

满足条件的区间分三种情况，区间的右端点是x，区间的左端点是x，或者x在区间的里面(且既不是区间的左端点也不是区间的右端点)，把这三种情况求和即可，我们先找到x，然后向两边去维护两个数组L[q]和R[q]，其值为有q个数比他少的情况有多少种，对于中间那种情况，枚举一遍即可，这样会有一个小问题，当k为1的时候，ans要加1，因为自己一个数也算一种情况了，特判就好

代码：

#include<bits/stdc++.h>
using namespace std;

#define ll long long
const int N = 2e5 + 5;
int n, k, x, a[N],L[N],R[N];

int main(){
    int cnt = 0;
    cin >> n >> x >> k;
    for (int i = 1; i <= n; i++){
        cin >> a[i];
        if (a[i] == x) cnt = i;
    }

    ll ans = 0, q = 0;
    for (int i = cnt - 1; i >= 1; i--){
        if (a[i] < x)  q++;
        L[q]++;
        if (q == k - 1) ans++;
    }

    q = 0;
    for (int i = cnt + 1; i <= n; i++){
        if (a[i] < x)  q++;
        R[q]++;
        if (q == k - 1) ans++;
    }

    for (int i = 0; i <= k - 1; i++){
        ans += 1ll * L[i] * R[k - 1 - i];
    }
    if (k == 1) ans ++;
    cout << ans << endl;
    return 0;
}