题目
给定三个整数数组
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≤100000,
0≤Ai,Bi,Ci≤100000
题解
思路:
这是枚举类题目,所以要考虑如何做到不重不漏,同时,如果用暴力的算法,三重循环的嵌套会导致O(n^3)的时间复杂度。这个问题我们可以将A和C分开,它们又都和B有关系,所以如果确定了B,A和C都可以确定。这样只需要枚举一次,降低时间复杂度。具体的实现方法是前缀和:用cnt[a[i]]++来存a[i]里的数字出现了多少次,s[i]为每个cnt[a[i]]的值]出现了多少次,求a[i]中比B[j]小的数==求从0到b[i]-1中有多少个数,所以A中比B小的数为s[b[i]-1].其中,s[i]是按cnt[i]顺序加的,保证了从小到大加,同理可得c[k]中比B[j]大的==全部数-去C中比B小的,最后,将比B小的方案数乘以比B大的方案数。