第九届蓝桥杯B组第六题递增三元组

递增三元组

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

思路分析

要找递增三元组，首先想到一个方法：暴力枚举，枚举A,B,C然后判断，但这样的时间复杂度是O(n^3),对于100000肯定是无法承受的。
那么我们不妨从递增入手，不难想到将A,B,C数组排序，当A,B确定后，只要找到第一个比B大的C的位置，就可以利用下标算出有多少个C可以选择，这样复杂度降到O(n^2)
这之后其实还可以继续优化。每个B可以对应的C的个数其实是一样的，因此可以用一个dp数组记录选择B可以组成的三元组个数，这样只需要完整搜索一次A,B,C，又可以降低复杂度

代码：

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
typedef long long LL;
const int MAXN = 100000;
int N, A[MAXN + 10], B[MAXN + 10], C[MAXN + 10];
int dp[MAXN + 10];	//记录以B[i]数为第二个可以组成的递增三元组 
int main()
{
	memset(dp, -1, sizeof(dp));
	scanf("%d", &N);
	for(int i = 1; i <= N; i++){
		scanf("%d", &A[i]);
	}
	for(int i = 1; i <= N; i++){
		scanf("%d", &B[i]);
	}
	for(int i = 1; i <= N; i++){
		scanf("%d", &C[i]);
	}
	sort(A, A + N);
	sort(B, B + N);
	sort(C, C + N);
	int ans = 0;
	//运用dp将复杂度降至O(n) 
	for(int i = 1; i <= N; i++){
		int k = 1;
		for(int j = 1; j <= N; j++){	//以B[j]可形成的三元组 
			if(B[j] <= A[i])
				continue;
			if(dp[j] != -1){	//如果dp存在 
				ans += dp[j];
				continue;
			}
			int cnt = 0;
			for(; k <= N; k++){ 
				if(C[k] <= B[j])
					continue;
				cnt = N - k + 1;
				break;
			}
			dp[j] = cnt;
			ans += cnt;
		}
	}
	printf("%d\n", ans);
 	return 0;
}