思路:
如果直接暴力 三个 for, O(n^3) 超时,所以优化。
关键是b 它是中间值,找出a []中比它小的数的个数as,找出b[]中比它大的数的个数cs,as * cs即为答案
分析图:
代码:
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int N=100010;
typedef long long LL;
int n,a[N],b[N],c[N];
int as[N];//as[N] 表示 A[] 中有多少个小于 b[i]
int cs[N];//cs[N] 表示 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]++;//都加 1 是因为:前缀和计算时,数组下标从 1 开始
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]]++;//再用前缀和时,这也是另一种排序,这个只需要 O(n) ,而其他的排序一般都是 O(n^2)
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]; // 求cnt[]的前缀和
for(int i=0;i<n;i++) cs[i]=s[N-1]-s[b[i]];
LL res=0;
for(int i=0;i<n;i++) res+=(LL)as[i]*cs[i];//因为得出的答案可能会爆 int 所以加上 LL, 强制类型转换
cout<<res<<endl;
return 0;
}