【每日一题】补档 CF1538C. Number of Pairs | 二分 | 简单

题目内容

原题链接

给定一个长度为 $n$ 的数组 $a$ ，找到有多少不同的下标对 $(i,j),i\ne j$ ，满足 $l\le a_i+a_j\le r$ 。

数据范围

• $1\le n\le 2\cdot 10^5$
• $1\le a_i\le 10^9$
• $1\le l\le r\le 10^9$

题解

首先下标只要不同即可，所以先对数组排序。

枚举下标 $i$ ，对于数 $a_i$ ，

• 找到 $j_1>i$ 的最小的 $j_1$ ，满足 $a_i+a_{j_1}\ge l$
• 找到 $j_2>i$ 的最大的 $j_2$ ，满足 $a_i+a_{j_2}\le r$

那么对于 $a_i$ ，就有 $j_2-j_1+1$ 个下标 $j$ 使得 $l\le a_i+a_j\le r$

当然后面的部分也可以使用双指针，不过时间复杂度的瓶颈在于排序。

时间复杂度：$O(n\log n)$

代码

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

void solve() {
    int n, x, y;
    cin >> n >> x >> y;

    vector<int> a(n);
    for (int i = 0; i < n; ++i) cin >> a[i];

    sort(a.begin(), a.end());

    long long ans = 0;
    for (int i = 0; i < n - 1; ++i) {
        if (a[i] + a[n - 1] < x) continue;
        if (a[i] + a[i + 1] > y) continue;

        int l = i + 1, r = n - 1;
        while (l < r) {
            int mid = (l + r) >> 1;
            if (a[i] + a[mid] >= x) r = mid;
            else l = mid + 1;
        }

        int left = l;

        l = i + 1, r = n - 1;
        while (l < r) {
            int mid = (l + r + 1) >> 1;
            if (a[i] + a[mid] <= y) l = mid;
            else r = mid - 1;
        }

        ans += r - left + 1;
    }

    cout << ans << "\n";
}

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    int T = 1;
    cin >> T;
    while (T--) {
        solve();
    }

    return 0;
}