递增三元组

递增三元组

难度：中等

题目

给定三个整数数组

A=[A₁,A₂,…A_N],
B=[B₁,B₂,…B_N],
C=[C₁,C₂,…C_N],

请你统计有多少个三元组 (i,j,k) 满足：

1. 1≤i,j,k≤N
2. A_i<B_j<C_k

输入格式

第一行包含一个整数 N。

第二行包含 N 个整数 A₁,A₂,…A_N。

第三行包含 N 个整数 B₁,B₂,…B_N。

第四行包含 N 个整数 C₁,C₂,…C_N。

输出格式

一个整数表示答案。

数据范围

1≤N≤10⁵,
0≤A_i,B_i,C_i≤10⁵

输入样例：

3
1 1 1
2 2 2
3 3 3

输出样例：

27

分析

枚举的要点：1. 顺序 2. 优化

思路

1. 暴力做法
   

2. 优化做法

   （1）先枚举B：

   题目中的数据范围是10⁵⇨时间复杂度是O(n) / O(nlogⁿ)⇨所以时间复杂度一定是小于nlogⁿ⇨最多枚举一个数组

   由题意 A_i<B_j<C_k

   如果先枚举A，会发现B和C不是独立的，是相互限制的（A和C都在两边，两者情况等价）

   如果两个变量不是完全独立的，那么我们在统计数量的时候，就不能简单的使用乘法原理相乘，比较麻烦。所以，应该先枚举B。因为枚举完以后，A和C的取值只和B有关系，它们之间没有任何限制的，因此它们是完全独立的，则在计算方案数的时候，只需要分别求一下有多少个A满足要求，有多少个C满足要求，相乘则行了。

   （2）对于每个B_j :

   1. 在A中有多少个小于B_j
   2. 在C中有多少个大于C_j

​

前缀和做的前提：题目中每一个数值都是小于等于10⁵

AC代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>

using namespace std;

typedef long long ll;

const int N=100010;

int n;
int a[N],b[N],c[N];
int as[N];//as[i]表示在A[]中有多少个数小于b[i] 
int cs[N];//cs[i]表示在C[]中有多少个数大于b[i]
int cnt[N],s[N];

int main()
{
	scanf("%d",&n);
	for(int i=0;i<n;i++)
	{
		scanf("%d",&a[i]);
		a[i]++;
	}
	for(int i=0;i<n;i++)
	{
		scanf("%d",&b[i]);
		b[i]++;
	}
	for(int i=0;i<n;i++)
	{
		scanf("%d",&c[i]);
		c[i]++;
	}
	
	//求as[]
	for(int i=0;i<n;i++)
	{
		cnt[a[i]]++;
	} 
	for(int i=1;i<N;i++)
	{
		s[i]=s[i-1]+cnt[i];//求cnt[]前缀和 
	}
	for(int i=0;i<n;i++)
	{
		as[i]=s[b[i]-1];
	}
	
	//求cs[]
	memset(cnt,0,sizeof cnt);
	memset(s,0,sizeof s);
	for(int i=0;i<n;i++)
	{
		cnt[c[i]]++;
	} 
	for(int i=1;i<N;i++)
	{
		s[i]=s[i-1]+cnt[i];
	}
	for(int i=0;i<n;i++)
	{
		cs[i]=s[N-1]-s[b[i]];
	}
	
	//枚举每个b[i]
	ll res=0;
	for(int i=0;i<n;i++)
	{
		res+=(ll)as[i]*cs[i];
	} 
	
	cout<<res<<endl;
	
	return 0;
}