题目描述
给定三个整数数组
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
解法一:暴力枚举
- 枚举的方法无非就是遍历求解森林的所有节点,代码比较简单
#include<iostream>
#include<algorithm>
using namespace std;
const int maxsize = 100000;
int main() {
int a[maxsize+10], b[maxsize+10], c[maxsize+10];
int n;
while (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];
long long sum = 0;
for (int i = 0; i < n; i++) {
for (int j = 0; j < n; j++) {
for (int k = 0; k < n; k++) {
if (a[i] < b[i] && b[i] < c[i]) {
sum++;
}
}
}
}
cout << sum << endl;
}
}
- 但是这种方法时间复杂度太高(O(n^3)),下面是基于二分查找的一点优化。
解法二:二分查找
- 代码如下:
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn = 1e6 + 10;
int a[maxn], b[maxn], c[maxn];
int binarySearch(int a[], int len, int key) {
//返回第一个大于关键字的索引
int l = 0, r = len - 1, mid, pos = -1;
while (l <= r) {
mid = (l + r) / 2;
if (a[mid] <= key) {
l = mid + 1;
pos = l;
} else {
r = mid - 1;
pos = mid;
}
}
return pos;
}
int main() {
int n;
while (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);
long long sum = 0;
// 2019.3.2前错解
/*for (int i = 0; i < n; i++) {
int bstart = binarySearch(b, n, a[i]);
int cstart = binarySearch(c, n, b[bstart]);
sum += (n - bstart) * (n - cstart);
}*/
for (int i = 0; i < n; i++) {
int bstart = binarySearch(b, n, a[i]);
for (int j = bstart; j < n; j++) {
int cstart = binarySearch(c, n, b[j]);
sum += n - cstart;
}
}
cout << sum << endl;
}
return 0;
}