第九届蓝桥杯真题：递增三元组

题目描述

给定三个整数数组
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;
}