给定三个整数数组
A=[A1,A2,…AN],
B=[B1,B2,…BN],
C=[C1,C2,…CN],请你统计有多少个三元组 (i,j,k)
满足:
1≤i,j,k≤N
Ai<Bj<Ck
输入格式
第一行包含一个整数 N。第二行包含 N个整数 A1,A2,…AN。第三行包含 N个整数 B1,B2,…BN。
第四行包含 N个整数 C1,C2,…CN。
输出格式
一个整数表示答案。
数据范围
1≤N≤105,0≤Ai,Bi,Ci≤105
输入样例:
3
1 1 1
2 2 2
3 3 3
输出样例:
27
本题的关键在于发现每个递增三元组中a[i]和c[k]都要受b[j]约束,故应该通过枚举b[j]计算三元组的个数。同时本题的易错点是统计各个值出现次数的时候,最后一个元素的下标由数据最大值决定,统一用数据范围最大值就好。
AC代码:
#include<stdio.h>
#include<string.h>
const int N=100010;
typedef long long LL;
int n;
int a[N],b[N],c[N];
int cnt[N],s[N];
int anum[N],cnum[N];
int main()
{
scanf("%d",&n);
//因为要计算值出现次数的前缀和,把所有值+1避免占用0下标
for(int i=1;i<=n;++i){scanf("%d",&a[i]);++a[i];}
for(int i=1;i<=n;++i){scanf("%d",&b[i]);++b[i];}
for(int i=1;i<=n;++i){scanf("%d",&c[i]);++c[i];}
//统计a数组中各个值出现的次数
for(int i=1;i<=n;++i) ++cnt[a[i]];
//计算a数组中各个值出现次数的前缀和
//因为值的范围是1~100001,cnt的最后一项为cnt[100001]
for(int i=1;i<N;++i) s[i]=s[i-1]+cnt[i];
//anum[i]中保存a中小于b[i]的数的个数
for(int i=1;i<=n;++i) anum[i]=s[b[i]-1];
memset(cnt,0,sizeof(cnt));
memset(s,0,sizeof(s));
for(int i=1;i<=n;++i) ++cnt[c[i]];
for(int i=1;i<N;++i) s[i]=s[i-1]+cnt[i];
//cnum[i]中保存c中大于b[i]的数的个数
for(int i=1;i<=n;++i) cnum[i]=s[N-1]-s[b[i]];
LL ans=0;//统计答案
for(int i=1;i<=n;++i) ans+=(LL)anum[i]*cnum[i];
printf("%lld",ans);
return 0;
}