当三个国家中的任何两国实力之和都大于第三国的时候,这三个国家互相结盟就呈“三足鼎立”之势,这种状态是最稳定的。
现已知本国的实力值,又给出 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}。
考了一个两边之和大于第三边,具体思路是这样的:
如果直接双循环判断,时间复杂度O(n方),是一定会超时的,于是整个题就变成了如何降低时间复杂度的问题,我们可以不用暴力法,改用二分法查找,具体思路:
先将数组整个排序,按照从大到小的顺序,然后遍历,将这个数固定为选的一条边,这样就有同时的两条边,现在只需要确定第三条边就行,且要满足约束条件,我们在i+1到n的这些树里选,如果能够满足大于两个之和,且小于两边只差,那么这两个数中间的那些数就是能够构成三角形的,图解法演示:
现在我们需要做的就是二分法找到大于最小边界的数和小于最大边界的数即可,然后下标相减就是所要的
#include<iostream>
#include<string>
#include<algorithm>
#include<bits/stdc++.h>
#include<stack>
#include<set>
#include<vector>
#include<map>
#include<queue>
#include<deque>
#include<cctype>
#include<unordered_set>
#include<unordered_map>
#include<fstream>
using namespace std;
int main(){
int n,p;
cin>>n>>p;
int a[100001];
int sum=0;
for(int i=1;i<=n;i++){
cin>>a[i];
}
sort(a+1,a+n+1);
for(int i=1;i<=n;i++){
int l=i+1,r=n,minn=abs(a[i]-p),maxx=a[i]+p;//l是左边界,r是右边界,minn是最小值,maxx是最大值
int num1,num2;//记录下标位置
if(l>r) break;
while(l<=r){//找到大于最小值的最大的数
int mid=(l+r)/2;
if(a[mid]<=minn){
l=mid+1;
}
else{
r=mid-1;
}
}
num1=l;
l=i+1,r=n;
while(l<=r){//找到小于最大值的最小的数
int mid=(l+r)/2;
if(a[mid]<maxx){
l=mid+1;
}
else{
r=mid-1;
}
}
num2=r;
sum+=(num2-num1+1);
}
cout<<sum;
return 0;
}