7-8 三足鼎立 (25分)(bound的使用)

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

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

输入格式:
输入首先在第一行给出 2 个正整数 n(2≤n≤1e5)和 P(≤1e9),分别为其他国家的个数、以及本国的实力值。随后一行给出 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 
#include 
using namespace std;

const int N = 1e5 + 10;
typedef long long LL;
int a[N];
int n, s;

int main()
{
    scanf("%d%d", &n, &s);
    for(int i = 1; i <= n; i ++ ) scanf("%d", &a[i]);

    sort(a + 1, a + n + 1);    
    LL ans = 0;
    for(int i = 1; i <= n; i ++ )
    {
        //首先找到不满足A + B > C的,然后找到abs(A - B) < C
        //其次是要注意从当前的i位置开始搜索!因为避免重复!
        int q = lower_bound(a + 1 + i, a + n + 1, s + a[i]) - (a + 1);
        int p = upper_bound(a + 1 + i, a + n + 1, abs(s - a[i])) - (a + 1);
        ans += q - p;
    }
    
    cout << ans << endl;
    return 0;
}

你可能感兴趣的:(bound的使用)