三足鼎立(二分法,简洁AC)

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

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

输入格式:

输入首先在第一行给出 2 个正整数 n(2≤n≤105)和 P(≤109),分别为其他国家的个数、以及本国的实力值。随后一行给出 n 个正整数,表示n 个其他国家的实力值。每个数值不超过 109,数字间以空格分隔。

输出格式:

在一行中输出本国结盟选择的个数。

输入样例:

7 30
42 16 2 51 92 27 35

样例:">输出样例:

9

ps 刚开始直接俩个for循环枚举结果只有16分,其他都超时,于是改进一下用二分过了

#include<iostream>
#include <algorithm>
using namespace std;
int main()
{
    long n,p,a[100001],num=0,head=0;
    cin >> n >> p;
    for(long z=0;z<n;z++) cin >> a[z];
    sort(a,a+n);
    /*   开始核心判断   */
    for(long z=0;z<n;z++)
    {
        long flag1 = abs(p-a[z]),flag2=p+a[z],p1=z+1,p2=n-1;
        //  二分找到符合要求的头部
        while (p1<=p2)
        {
            long avg = (p1+p2+1)/2;
            if(a[avg]>flag1) p2=avg-1;
            else p1 = avg+1;
        }
        flag1 = p1;
        //  二分找到符合要求的结尾
        p1=z+1,p2=n-1;
        while (p1<=p2)
        {
            long avg = (p1+p2+1)/2;
            if(a[avg]>=flag2) p2=avg-1;
            else p1 = avg+1;
        }
        num += p2-flag1+1;
    }
    //  答案输出
    cout << num;
    return 0;
}
经验分享 程序员 微信小程序 职场和发展