当三个国家中的任何两国实力之和都大于第三国的时候,这三个国家互相结盟就呈“三足鼎立”之势,这种状态是最稳定的。
现已知本国的实力值,又给出 n 个其他国家的实力值。我们需要从这 n 个国家中找 2 个结盟,以成三足鼎立。有多少种选择呢?
输入格式:
输入首先在第一行给出 2 个正整数 n(2≤n≤105)和 P(≤10e9),分别为其他国家的个数、以及本国的实力值。随后一行给出 n 个正整数,表示n 个其他国家的实力值。每个数值不超过 10e9,数字间以空格分隔。
输出格式:
在一行中输出本国结盟选择的个数。
输入样例:
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}。
知识引入
AC本道题需要用到C++ STL中的lower_bound()和upper_bound()函数。
lower_bound(type a[ ], type* p1, type* p2)
能够查找第一个大于等于当前数字的地址
upper_bound(type a[ ], type* p1, type* p2)
能够查找第一个大于(没有等于)当前数字的地址
思路
根据题目定义可知任意两个国家的实力值要大于第三个。如果直接双循环搜索,必然超时(不过用来骗分不错)
让我们想想,假如把各个国家的实力值进行排序,依次遍历数组,如果要满足当前的实力值和题目给定的实力值三足鼎立,那么第三个国家的实力值的范围可以确定下来的。
我们不妨这样想,设当前实力值加上输入给定的实力值为temp1,当前实力值减去输入给定的实力值的绝对值为temp2,在当前位置后找第一个大于等于temp的数和第一个大于temp2的数。当前遍历的国家可以和两个位置数之间的国家形成联盟,每次遍历总数加上两数之差,最后输出总是即可。
参考代码
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int num[100005];int main()
{int n, p;scanf("%d %d", &n, &p);ll sum = 0;for(int i = 1; i <= n; i++)scanf("%d", &num[i]);sort(num+1, num+n+1);int k, q;for(int i = 1; i <= n; i++){q = lower_bound(num+i+1, num+n+1, p+num[i]) - (num+1);k = upper_bound(num+i+1, num+n+1, abs(num[i]-p)) - (num+1);sum += q-k;}printf("%ld\n", sum);return 0;
}