L4-203 三足鼎立（c++，python）

汇太浪

已于 2024-06-05 13:20:57 修改

阅读量488

点赞数 4

文章标签： python c++

于 2024-06-03 20:29:17 首次发布

本文链接：https://blog.csdn.net/zjl_hui/article/details/139423473

版权

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

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

输入格式：

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

输出格式：

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

输入样例：

7 30
42 16 2 51 92 27 35

输出样例：

样例解释：

能联合的另外 2 个国家的 9 种选择分别为：

{16, 27}, {16, 35}, {16, 42}, {27, 35}, {27, 42}, {27, 51}, {35, 42}, {35, 51}, {42, 51}。

分析：

必须用二分查找，否则会超时，c++里的算法库algorithm为我们提供了upper_bound和lower_bound，分别用来找一个排好序的数组中指定区间[begin,end)中第一个大于和大于等于某个数的位置，找到返回该数字的地址，不存在则返回end，通过返回的地址减去起始地址begin,得到找到数字在数组中的下标。

先按照各国的武力值排序，然后遍历每个国家，求出第三个国家武力值的区间（已知两个三角形两条边，求出第三条边的区间），然后用二分查找找到左右端点的坐标l和r（注意l要用upper_bound找，这样从l开始的元素就能满足大于另外两个数差值的绝对值，r要用lower_bound找，这样r之前的元素就能满足小于另外两个数的和）

c++代码：

#include"iostream"
#include"algorithm"
#include"math.h"
using namespace std;
int main(){
    long long n,p;
    cin>>n>>p;
    long long nums[100005];
    for(int i=0;i<n;i++){
        cin>>nums[i];
    }
    sort(nums,nums+n);
    long long ans = 0;
    for(int i=0;i<n;i++){
        int l = upper_bound(nums+i+1,nums+n,abs(p-nums[i])) - nums;
        int r = lower_bound(nums+i+1,nums+n,p+nums[i]) - nums;
        ans += r - l;
    }
    cout<<ans;
}

同样的，python中的二分查找库bisect为我们提供了bisect_left和bisect_right，对应c++的lower_bound和upper_bound，光写一个bisect作用等同于bisect_right

但是这题我写的python代码AC不了，pypy3内存会爆炸，python3则会超时：

import bisect
n, p = map(int,input().split())
nums = list(map(int,input().split()))
nums.sort()
ans = 0
for i in range(n):
    ans += bisect.bisect_left(nums[i+1:],nums[i]+p) - bisect.bisect_right(nums[i+1:],abs(nums[i]-p))
print(ans)