给定三个整数数组
A = [A1, A2, ... AN],
B = [B1, B2, ... BN],
C = [C1, C2, ... CN],
请你统计有多少个三元组(i, j, k) 满足:
1. 1 <= i, j, k <= N
2. Ai < Bj < Ck
【输入格式】
第一行包含一个整数N。
第二行包含N个整数A1, A2, ... AN。
第三行包含N个整数B1, B2, ... BN。
第四行包含N个整数C1, C2, ... CN。
对于30%的数据,1 <= N <= 100
对于60%的数据,1 <= N <= 1000
对于100%的数据,1 <= N <= 100000 0 <= Ai, Bi, Ci <= 100000
【输出格式】
一个整数表示答案
【样例输入】
3
1 1 1
2 2 2
3 3 3
【样例输出】
27
资源约定:
峰值内存消耗(含虚拟机) < 256M
CPU消耗 < 1000ms
分析:对于任意一个b[i],a中小于b[i]的个数*c中大于b[i]的个数就是当前b[i]所能构造的所有的递增三元组。
那么快速求出比b[i]小的a元素个数,比b[i]大的c元素个数即可。
#include<bits/stdc++.h>
using namespace std;
const int MAXN = 100005;
int a[MAXN], b[MAXN], c[MAXN];
int n, sum;
int main()
{
cin >> n;
for (int i = 0; i < n; i++)cin >> a[i];
for (int i = 0; i < n; i++)cin >> b[i];
for (int i = 0; i < n; i++)cin >> c[i];
sort(a, a + n);
sort(b, b + n);
sort(c, c + n);
sum = 0;
for (int i = 0; i < n; i++) {
int x = (lower_bound(a, a + n, b[i]) - a);//a中有多少个比b[i]小的
int y = (n - (upper_bound(c, c + n, b[i]) - c));//c中有多少个比b[i]大的
sum += x * y;
}
cout << sum;
return 0;
}