题目描述
给定三个整数数组
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
代码:(还没AC超时)
#include<bits/stdc++.h>
using namespace std;
int main()
{
long long count=0;
int N;
cin>>N;
int a[100009],b[100009],c[100009];
int mina;
for(int i=1;i<=N;i++) cin>>a[i];
for(int i=1;i<=N;i++) cin>>b[i];
for(int i=1;i<=N;i++) cin>>c[i];
sort(a,a+N+1);
sort(b,b+N+1);
sort(c,c+N+1);
for(int j=1;j<=N;j++){
for(int i=N;i>0;i--)
{
if(b[j]>a[i]){
mina=i; //bj小的a的元素的数量为i
for(int k=1;k<=N;k++){
if(c[k]>b[j]){
count+=(N-k+1)*mina;//bj大的c的元素的数量为N - k+1
break;
}
}
break;
}
}
}
cout<<count;
return 0;
}
思路:对于a,b,c数组进行排序。枚举每一个bj,然后去判断ai和bj,bj和ck的大小,算出比bj小的a数组里的数量i和比bj大的c数组里的数量N-k+1相乘即为一个bj所有的三元组。
(目前还是超时,要使用二分才能保证AC,明天弄= =)