题目
分析
直接暴力 O ( N 3 ) O(N^3) O(N3),显然超时。
将三个数组排序,遍历 b b b数组,二分找到 a a a中小于 b [ i ] b[i] b[i]的个数 A A A,找到 c c c中大于 b [ i ] b[i] b[i]的个数 C C C, a n s ans ans+= A ∗ C A*C A∗C。
时间复杂度,排序 O ( N log N ) O(N \log N) O(NlogN),查找 O ( N log N ) O(N \log N) O(NlogN),总体 O ( N log N ) O(N \log N) O(NlogN)。
代码
自己写二分
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long LL;
const int N = 100010;
int n;
int a[N], b[N], c[N];
LL ans;
int main()
{
scanf("%d", &n);
for (int i = 0; i < n; i++) scanf("%d", &a[i]);
for (int i = 0; i < n; i++) scanf("%d", &b[i]);
for (int i = 0; i < n; i++) scanf("%d", &c[i]);
sort(a, a + n);
sort(b, b + n);
sort(c, c + n);
/* 二分:
* 1. 找边界
* 2. 求mid,写check函数
* 3. 若l = mid则 +1,r = mid不用 +1
*
* 需要注意的是,二分出来的点若在边界则需要特判。
*/
for (int i = 0; i < n; i++)
{
int A = 0, C = 0;
int l = 0, r = n - 1;
while (l < r)
{
int mid = l + r + 1 >> 1;
if (a[mid] < b[i]) l = mid;
else r = mid - 1;
}
if (a[l] < b[i]) A = l + 1;
l = 0, r = n - 1;
while (l < r)
{
int mid = l + r >> 1;
if (c[mid] > b[i]) r = mid;
else l = mid + 1;
}
if (c[l] > b[i]) C = n - l;
ans += (LL)A * C;
}
cout << ans << endl;
return 0;
}