牛客练习赛87.B-k小数查询(如果绝对思路没问题,请换种写法,换种更简单的写法)

总结&收获

  1. 如果绝对思路没问题,请换种写法,换种更简单的写法。
  2. 太久没写题了,手真的生疏emm

题目

  1. 传送门
  2. 题意:注意a[]是一个排列
    在这里插入图片描述
    在这里插入图片描述
  3. 题解
    1. 枚举pos前面<x的个数和后面<x的个数即可。
    2. os:当然需要提前处理一下数据。
  4. 代码
//如果绝对思路没问题,请换种写法,换种更简单的写法
#include <bits/stdc++.h>
#define int long long
#define dbg(x) cout << #x << "===" << x << endl
using namespace std;
const int N = 2e5 + 5;
int n, k, x;
int a[N];
int L[N], R[N];
signed main() {
    
    
    cin >> n >> x >> k;
    int pos;
    for (int i = 1; i <= n; i++) {
    
    
        cin >> a[i];
        if (a[i] == x) pos = i;
    }
    int cnt = 0;
    for (int i = pos; i >= 1; i--) {
    
    
        if (a[i] < x)
            L[++cnt]++;
        else
            L[cnt]++;
    }
    cnt = 0;
    for (int i = pos; i <= n; i++) {
    
    
        if (a[i] < x)
            R[++cnt]++;
        else
            R[cnt]++;
    }
    int ans = 0;
    for (int i = 0; i <= k - 1; i++) ans += L[i] * R[k - 1 - i];
    cout << ans << endl;
    return 0;
}

猜你喜欢

转载自blog.csdn.net/I_have_a_world/article/details/119836444