当三个国家中的任何两国实力之和都大于第三国的时候,这三个国家互相结盟就呈“三足鼎立”之势,这种状态是最稳定的。
现已知本国的实力值,又给出 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}。
用二分法优化多重循环找不同的国家
#include<bits/stdc++.h>
using namespace std;
int a[100010];
bool cmp(int a,int b)//sort函数运用,用来从小到大排序
{
return a>b;
}
int main(){
int n,p,left,right,mid,flag = 0,tmp;
long long cnt = 0;//找的数
cin>>n>>p;
for(int i = 0;i<n;i++)
cin>>a[i];
sort(a,a+n,cmp);//快速排序
for(int i = 0;i<n;i++){
left = i+1;
right = n-1;
if(a[left]+p<=a[right]) continue;//快速判断,如果此时的第三边(此时的第三边是最大的)过小,那就不用继续了
else{
while(left<=right){
mid = (left+right)/2;
if(a[i]+a[mid]>p&&a[i]+p>a[mid]&&a[mid]+p>a[i]){//如果此时mid符合,则找mid左边的数
tmp = mid;
left = mid+1;
flag = 1;
}
else right = mid-1;
}
}
if(flag == 1){
cnt+=tmp-i;
}
flag = 0;
}
printf("%lld\n",cnt);
return 0;
}