题目
给定三个整数数组
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。
对于30%的数据,1 <= N <= 100
对于60%的数据,1 <= N <= 1000
对于100%的数据,1 <= N <= 100000 0 <= Ai, Bi, Ci <= 100000
【输出格式】
一个整数表示答案
【输入样例】
3
1 1 1
2 2 2
3 3 3
【输出样例】
27
资源约定:
峰值内存消耗(含虚拟机) < 256M
CPU消耗 < 1000ms
请严格按要求输出,不要画蛇添足地打印类似:“请您输入…” 的多余内容。
所有代码放在同一个源文件中,调试通过后,拷贝提交该源码。
不要使用package语句。不要使用jdk1.7及以上版本的特性。
主类的名字必须是:Main,否则按无效代码处理。
思路:前缀和
这道题暴力很好想,但是时间肯定超时。这道题我一开始用的二分法,但是还得先将数组排序,然后遍历B数组,将A数组满足的元素个数乘C数组满足的元素个数。思路也很简单,但是哈哈我没写出来,二分掌握的不好。
这个题用前缀和也是十分好用的,前缀和的特点就是在数据输入阶段就将数据处理了。为计算阶段提供前缀和数组,辅助计算。二分是算出下标,然后用N-下标就是满足的个数。前缀和是直接输入的时候将每个数出现的次数存起来,然后形成前缀和数组。遍历B数组的时候直接取过来用就可以了。
还有就是没想到最后一个测试点的结果连long都会溢出,最后只能改成Bigdecimal。
static int[] a;
static int[] b;
static long[] ca;
static long[] cc;
static int[] c;
public static void main(String args[]) {
Scanner in=new Scanner(System.in);
int N=in.nextInt();BigDecimal ans=new BigDecimal(0);
a=new int[N];
b=new int[N];
c=new int[N];
ca=new long[1000001];
cc=new long[1000001];
for(int i=0;i<N;i++)
{
a[i]=in.nextInt();
ca[a[i]]++; //前缀和的特点就行在输入阶段就将数据处理了
}
for(int i=0;i<N;i++)
b[i]=in.nextInt();
for(int i=0;i<N;i++)
{
c[i]=in.nextInt();
cc[c[i]]++;
}
for(int i=1;i<1000001;i++)
ca[i]+=ca[i-1];
for(int i=1000000;i>0;i--)
cc[i-1]+=cc[i];
for(int i=0;i<N;i++)
if(b[i]!=0)
ans=ans.add(new BigDecimal(ca[b[i]-1]*cc[b[i]+1]));
System.out.println(ans);
}
思路参考:https://www.bilibili.com/video/BV1u5411o7qR?share_source=copy_web