时间限制: 1Sec 内存限制: 128MB
题目描述
给定三个整数数组
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。
输出
一个整数表示答案
样例输入
3
1 1 1
2 2 2
3 3 3
样例输出
27
提示
对于30%的数据,1 <= N <= 100 对于60%的数据,1 <= N <= 1000 对于100%的数据,1 <= N <= 100000 0 <= Ai, Bi, Ci <= 100000
分析:用普通方法肯定会超时,将A,B,C三个数组排序,对于每一个B[j],找到比它小的A[i]的数目和比它大的C[i]的数目,然后相乘,将所有的情况相加即可。注意:前面已经比b[j-1]小的数目已经找出来,那么就继续往后找就行,不需要从0开始
#include<bits/stdc++.h>
using namespace std;
int N,A[100005],B[100005],C[100005];
int main(){
cin>>N;
for(int i=0;i<N;i++){
scanf("%d",&A[i]);
}
for(int i=0;i<N;i++){
scanf("%d",&B[i]);
}
for(int i=0;i<N;i++){
scanf("%d",&C[i]);
}
sort(A,A+N);
sort(B,B+N);
sort(C,C+N);
long long left=0,right=0, sum=0;
for(int i=0;i<N;i++){//遍历B数组
while(left<N&&A[left]<B[i])left++;
while(right<N&&C[right]<=B[i])right++; //right表示C数组<=B[i]的数目
sum+=left*(N-right);
}
cout<<sum<<endl;
return 0;
}