L4-203 三足鼎立

当三个国家中的任何两国实力之和都大于第三国的时候，这三个国家互相结盟就呈“三足鼎立”之势，这种状态是最稳定的。

现已知本国的实力值，又给出 n 个其他国家的实力值。我们需要从这 n 个国家中找 2 个结盟，以成三足鼎立。有多少种选择呢？

输入格式：

输入首先在第一行给出 2 个正整数 n（2≤n≤105）和 P（≤109），分别为其他国家的个数、以及本国的实力值。随后一行给出 n 个正整数，表示n 个其他国家的实力值。每个数值不超过 109，数字间以空格分隔。

输出格式：

在一行中输出本国结盟选择的个数。

输入样例：

7 30
42 16 2 51 92 27 35

输出样例：

9

样例解释：

能联合的另外 2 个国家的 9 种选择分别为：

{16, 27}, {16, 35}, {16, 42}, {27, 35}, {27, 42}, {27, 51}, {35, 42}, {35, 51}, {42, 51}。

 代码：

#include <iostream>
#include <algorithm>

using namespace std;

typedef long long ll;  // 使用long long类型以支持大整数处理

ll n, p, res;  // n表示其他国家的数量，p是本国的实力值，res用于统计有效的结盟组合数量
ll nums[200010];  // 数组存储其他国家的实力值，注意数组大小是根据问题约束设定的

int main() {
    // 输入处理
    cin >> n >> p;  // 读入国家数量和本国的实力值
    for (int i = 0; i < n; i++) cin >> nums[i];  // 读入其他国家的实力值到数组中

    // 对其他国家的实力值进行排序，以便高效地进行范围查询
    sort(nums, nums + n);

    for (int i = 0; i < n; i++) {
        // 'l' 是第一个使得nums[i]和nums[l]之和不能击败本国的索引位置
        int l = upper_bound(nums + i + 1, nums + n, p - nums[i]) - nums;

        // 'r' 是第一个使得nums[i]和nums[r]之和可以击败本国的索引位置
        int r = lower_bound(nums + i + 1, nums + n, p + nums[i]) - nums;

        res += (r - l);
    }
    // 输出最终的结果
    cout << res;
}