【GPLT 二阶题目集】L4-203 三足鼎立

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

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

输入格式：

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

输出格式：

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

输入样例：

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 <vector>
#include <algorithm>
using namespace std;
#define MAXSIZE 100010
#define ll long long

int main()
{
    int n;
    ll p, temp, res=0; cin >> n >> p;
    vector<ll> powers; //其他国家实力值
    for (int i = 0; i < n; i++) {
        cin >> temp;
        powers.push_back(temp);
    }
    sort(powers.begin(), powers.end());

    for (int i = 0; i < n; i++) {
        int up = p + powers[i], down = abs(p - powers[i]); //获取第三个国家实力值的上下边界
        //符合条件国家的下标范围：[first,last]
        int first = upper_bound(powers.begin() + i + 1, powers.end(), down) - powers.begin();
        int last = lower_bound(powers.begin() + i + 1, powers.end(), up) - powers.begin() - 1;
        if (first <= last) res += last - first + 1;
    }
    cout << res << endl;
    return 0;
}

 注意事项：

两边之和大于第三边，则 |a-b|<c<a+b 。

这题要注意不能运行超时，二分查找或者查找函数都可以用，不能理解的地方是为什么res不用long long变量，好几个测试点过不去？？！(18分)

如有问题，欢迎提出。