【蓝桥杯】递增三元组 (二分, lower_bound, upper_bound)

有三个数组A，B，C，每个数组中有n个数，你可以从每个数组中找一个数，使得Ai<Bj<Ck ,(1<=I,j,k<=n)(1<=n<=100000,1<=Ai,Bj,Ck<=1000000)，求最多可以组出多少三元组

Input

有多组输入
第一行输入n
接下来三行输入A，B，C三个数组，每个数组n个数

Output

每行一个整数，表示最多有多少三元组

Sample Input

3
1 1 1
2 2 2
3 3 3
3
1 2 3
1 2 3
1 2 3
Sample Output

27
1

思路分析：

看n的数据，时间复杂度只能在O(n)或者O(ngln)了。
其实仔细分析可以想到排序之后，可以用到二分查找。那么怎么设计呢？
一开始我是想的遍历a，这样b数组就得找第一个比a[i]大的数b[j]，c数组找第一个比b[j]大的数c[k]. 但是在草稿纸模拟一下发现根本不可行，b数组如果不遍历会把一些重复的元素算进去。因此我选择遍历b数组。
于是，在a数组找比b[j]小的索引i，在c数组找比b[j]大的索引k. 这样可以用lower_bound和upper_bound找出来，然后根据排列组合的乘法原理求出答案。
这里顺带介绍一下lower_bound和upper_bound用法。
lower_bound(a,a+n,k)表示在a数组[0,n)里找第一个大于等于k的值的迭代器。
upper_bound(a,a+n,k)表示b数组[0,n)里找第一个大于k的值的迭代器。
lower_bound(a,a+n,k)-a表示第一个大于等于k的值的索引。
distance(a,b)表示迭代器a和b之间的元素个数( [a,b) ). (注意a的位置要小于b，不然会报错).
distance的用处在于集合set里，set是一个平衡树，并不是连续的内存空间，想获取元素个数只能用distance.

#include <bits/stdc++.h>
using namespace std;

int main()
{
	int a[] = {1,2,2,3,3,5}, index_a, n = 6;
	index_a = lower_bound(a,a+n,2)-a;	//索引 1 
	cout<<index_a<<endl;
	index_a = distance(a, lower_bound(a,a+n,2));	//索引 1 
	cout<<index_a<<endl;
	index_a = upper_bound(a,a+n,2)-a;	//索引 3 
	cout<<index_a<<endl;
	index_a = distance(a, upper_bound(a,a+n,2));	//索引 3 
	cout<<index_a<<endl;
	index_a = lower_bound(a,a+n,5)-a;	//索引 5
	cout<<index_a<<endl;
	index_a = upper_bound(a,a+n,5)-a;	//超出范围，索引 6 
	cout<<index_a<<endl;
} 

介绍完lower_bound和upper_bound后，来分析一波。
我们遍历b数组，在a数组找比b[j]小的个数，在c数组找比b[j]大的个数.
第一个直接lower_bound(a.begin(), a.end(), b[i]) - a.begin();
lower_bound在a数组找到第一个比b[j]大于等于的数的索引，那么个数就是这个索引-a.begin().因为个数是在左闭右开的区间里，因为右开，所以区间里所有的数都比b[j]小。

第二个直接index_b = upper_bound(c.begin(), c.end(), b[i])- c.begin()求出第一个大于b[j]的数的索引，n-index_b就是索引区间[b,n)的个数。

最后，根据乘法原理，两个个数相乘。不过这两个值都可能在10⁵，相乘会爆int范围，这个要注意！！！不然就WA掉了，而且很难察觉！

另外，这道题不需要去重，不可思议。

下面上代码。

#include <iostream>
#include <vector>
#include <algorithm>
#define ll long long
#define MAX 100010
using namespace std;

int main()
{
	ll ans;
	int n, x;
	while(cin>>n)
	{
		ans = 0;
		vector<int> a, b, c;
		for(int i = 0;i < n;i++)
		scanf("%d",&x), a.push_back(x);
		for(int i = 0;i < n;i++)
		scanf("%d",&x), b.push_back(x);
		for(int i = 0;i < n;i++)
		scanf("%d",&x), c.push_back(x);
		
		sort(a.begin(), a.end());
		sort(b.begin(), b.end());
		sort(c.begin(), c.end());
		
		for(int i = 0;i < n;i++)
		{
//			if(b[i] == b[i-1]) continue;	不需要去重 
			ll index_a = lower_bound(a.begin(), a.end(), b[i]) - a.begin(); //索引a 
			ll index_c = upper_bound(c.begin(), c.end(), b[i])- c.begin();	//索引c 
//			cout<<index_a<<" "<<index_c<<endl;
			ans += index_a * (n-index_c);
		}
		cout<<ans<<endl;
	}
	return 0;
}