题目
当三个国家中的任何两国实力之和都大于第三国的时候,这三个国家互相结盟就呈“三足鼎立”之势,这种状态是最稳定的。
现已知本国的实力值,又给出 n 个其他国家的实力值。我们需要从这 n 个国家中找 2 个结盟,以成三足鼎立。有多少种选择呢?
输入
输入首先在第一行给出 2 个正整数 n ( 2 ≤ n ≤ 1 0 5 ) n(2≤n≤10^5) n(2≤n≤105)和 P ( ≤ 1 0 9 ) P(≤10^9) P(≤109),分别为其他国家的个数、以及本国的实力值。随后一行给出 n 个正整数,表示n 个其他国家的实力值。每个数值不超过 1 0 9 10^9 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}。
题目分析
本题类似于三角形,满足条件:两国相加大于第三国,两国相减小于第三国即可,那么对于选取任意两国后,在寻找满足条件的第三国时,可以计算出第三国的国力范围,选出在该范围内的国家即可。但通过题目的分析,数据量是巨大的,因此不可能逐一比较选取,此时便可以先将所得序列排序,然后对范围边界进行二分查找,找出对应下标,那么在下标范围内的国家都是可以选取的。
在二分查找中,用到的两个函数:
upper_bound(start,end,num)二分查找大于num的第一个数,并返回这个数的地址
lower_bound(start,end,num)二分查找大于等于num的第一个数,并返回这个数的地址
代码
#include<iostream>
#include<algorithm>
using namespace std;
int k[100005],m,n;
int main(){
long long int cnt=0;
cin >> n >> m;
for(int i = 0;i<n;++i) scanf("%d",&k[i]);
sort(k,k+n);
int l , r;
for(int i = 0;i<n;++i){
l = upper_bound(k+i+1,k+n,abs(m-k[i]))-k;
//二分查找确定地址并减去初始位置即为下标
r = lower_bound(k+i+1,k+n,m+k[i])-k;
cnt=cnt+(r-l);//计算符合条件的国家个数
}
cout << cnt<<endl;
return 0;
}