当三个国家中的任何两国实力之和都大于第三国的时候,这三个国家互相结盟就呈“三足鼎立”之势,这种状态是最稳定的。
现已知本国的实力值,又给出 n 个其他国家的实力值。我们需要从这 n 个国家中找 2 个结盟,以成三足鼎立。有多少种选择呢?
输入格式:
输入首先在第一行给出 2 个正整数 n(2≤n≤10e5)和 P(≤10e9),分别为其他国家的个数、以及本国的实力值。随后一行给出 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}。
思路:
分析一下数据量在10e5这个量级,暴力的时间复杂程度为O(N^2),肯定会超时.所以我们用二分查找,时间复杂程度O(NlogN).
首先用sort对实力值进行排序。接着当前在nums[i]满足的所有情况,我们可以使用二分查找,找到应该左边界与右边界,这两个位置之间的元素就是符合条件的组合数量。我们可以使用 upper_bound
找到第一个大于 fabs(p - nums[i])
的实力值的位置,然后再使用 lower_bound
找到第一个大于等于 p + nums[i]
的实力值的位置。
#include <iostream>
#include <algorithm>
#include <cmath>
using namespace std;
typedef long long ll;
ll n, p, nums[200010], res;
int main() {
cin >> n >> p;
for (int i = 0; i < n; i++)
cin >> nums[i];
// 对其他国家的实力值进行排序
sort(nums, nums + n);
for (int i = 0; i < n; i++) {
// 使用二分查找找到满足条件的组合数量
// upper_bound 找到第一个大于 fabs(p - nums[i]) 的位置
// lower_bound 找到第一个大于等于 p + nums[i] 的位置
int l = upper_bound(nums + i + 1, nums + n, fabs(p - nums[i])) - nums;
int r = lower_bound(nums + i + 1, nums + n, p + nums[i]) - nums;
res += (r - l);
}
cout << res << endl;
return 0;
}