递增三元组

题目描述
给定三个整数数组
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。
1 <= N <= 100000 0 <= Ai, Bi, Ci <= 100000
输出
一个整数表示答案
样例输入 Copy
3
1 1 1
2 2 2
3 3 3
样例输出 Copy
27

思路：最初我的思路就是三重循环，如果有Ai < Bj < Ck，则cnt++。
很明显，在O(N^3)的时间复杂度情况下，超时了。

优化思路：排序后使用二分算法查找，只需要进行一次循环，循环Bj，寻找Ai中最后一个比Bj小的下标，寻找Ck中第一个比Bj大的下标。最后将每次找到的数量相乘累加。

代码如下：

#include<iostream>
#include<algorithm>
using namespace std;

int n;
int a[10001],b[10001],c[10001];

//思路：
//在a[i]找到最后一个比b[i]小的下标
//在c[i]找到第一个比b[i]大的下标
 
int find1(int x)
{
	int l=0,r=n-1;
	while(l<r){
		int mid=(l+r+0.5)/2;			//确保mid每次取值数字不唯一时候选择较大的数 
		a[mid]<x?l=mid:r=mid-1;			//已经有了上一步，所以不需要选择 l=mid+1
	}
	return l;
}

int find2(int x)
{
	int l=0,r=n-1;
	while(l<r){
		int mid=(l+r)/2;				//确保mid每次取值数字不唯一时候选择较小的数 
		c[mid]>x?r=mid:l=mid+1;			//已经有了上一步， 所以不需要选择 r=mid-1 
	}
	return l;
}

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(c,c+n);
	
	long long res=0;
	
	for(int i=0;i<n;i++){
		int aa=find1(b[i]);
		int cc=find2(b[i]);
		if(a[aa]>=b[i]||c[cc]<=b[i]){
			continue;
		}
		
		res+=(long long)(aa+1)*(n-cc);
	}
	cout<<res;
	return 0;
}