[蓝桥杯2018初赛]递增三元组

题目描述

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

输出格式

一个整数表示答案

输入样例 复制

3
1 1 1
2 2 2
3 3 3

输出样例 复制

27

题目类型：模拟、组合

解题思路：枚举B，然后 用二分找第一个大于等于Bi 的Ai的位置，找第一个大于Bj的Ck的位置，再计算。

注意点：一直只过了88%的样例，后来发现， 在最终计算ans += （pa*（n-pc))时， pa和pc都要开long long 不然会爆。

AC代码：

#include <bits/stdc++.h>
using namespace std;
const int N =1e5+10;
int n;
int a[N], b[N], c[N];

int main()
{
	ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
	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 ans = 0;
	for(int j = 0; j< n; j++)
	{
		long long posa = lower_bound(a, a+n, b[j]) - a;
		long long posc =n -  (upper_bound(c, c+n, b[j]) - c);
		ans += (posa*posc);
	}
	cout<<ans<<endl;
	return 0;
}