k小数查询（二分的边界问题）

题目链接
边界的问题真的很烦。。又是写了一上午和一整晚的题。。
将排列中小于等于x的数标记为1。
这样通过查询区间 [ l, r ] 的和，就能判断是否有个数小于等于x。
可以用前缀和优化。
枚举每一个右端点r，查询有多少个左端点l满足 sum[l] == sum[r] - k。
因为区间一定要包含x，所以从x的位置pos开始枚举。
lower_bound查询的pos1是不能够>=pos的，因为pos1 == pos时，满足条件的 l 是 pos + 1，已经不包含x了。
upper_bound查询到的pos2是不能够>pos的，但是可以等于，这是因为upper_bound查询的是>sum[r] - k的第一个数，就算pos2==pos，那满足条件的 l 也是pos + 1 - 1 = pos，l 仍然包含了x。

#include <bits/stdc++.h>
using namespace std;
#define int long long
const int N = 200000 + 5, mod = 1e9 + 7;
int a[N], b[N], sum[N];
signed main()
{
    /* freopen("data5.in","r",stdin);
	freopen("data5.out","w",stdout); */
    int n, x, k;
    cin >> n >> x >> k;
    int p = 0;
    for (int i = 1; i <= n; i++) {
        cin >> a[i];
        if (a[i] == x) p = i;
        if (a[i] <= x) b[i] = 1;
    }

    int ans = 0;
    for (int i = 1; i <= n; i++) {
        sum[i] = sum[i - 1] + b[i];
        // cout << sum[i] << " ";
    }
    for (int i = p; i <= n; i++) {
        int pos1 = lower_bound(sum, sum + 1 + p, sum[i] - k) - sum;
        int pos2 = upper_bound(sum, sum + 1 + p, sum[i] - k) - sum;
        if (pos2 > p && pos1 < p) ans += p - pos1 - 1;
        else if (pos1 < p && pos2 <= p) ans += pos2 - pos1;
        // cout << "p1 = " << pos1 << ", p2 = " << pos2 << ", ans = " << ans << ", i = " << i << endl;
    }
    cout << ans;
    return 0;
}

O(N)的做法是，直接用桶来统计。
这里也有个小边界问题，if (i < p) sum[b[i]]++; 之所以不是 if(i <= p )，是因为，加了等于号之后，p + 1 位置也会被统计进ans，而p + 1位置当然是不包括x的，所以是 < 而不是 <=。

#include <bits/stdc++.h>
using namespace std;
#define int long long
const int N = 200000 + 5, mod = 1e9 + 7;
int a[N], b[N], sum[N];
signed main()
{
    /* freopen("data5.in","r",stdin);
	freopen("data5.out","w",stdout); */
    int n, x, k;
    cin >> n >> x >> k;
    int p = 0;
    for (int i = 1; i <= n; i++) {
        cin >> a[i];
        if (a[i] == x) p = i;
        if (a[i] <= x) b[i] = 1;
    }

    int ans = 0;
    for (int i = 1; i <= n; i++) {
        b[i] = b[i - 1] + b[i];
        if (i < p) sum[b[i]]++;
    }
    sum[0]++;
    for (int i = p; i <= n; i++) {
        if (b[i] - k < 0) continue;
        ans += sum[b[i] - k];
    }
    cout << ans;
    return 0;
}