给定三个整数数组
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数组当中找比b[i]小的数;
当然找完之后还要判断一下是否存在这个数即(a[l] <b[i])
在c数组当中找比b[i]大的数;
#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
const int N=100010;
int a[N];
int b[N];
int c[N];
int main(){
int n;
scanf("%d",&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(c,c+n);
long long cnt=0;
for(int i=0;i<n;i++){
int l=0;
int r=n-1;
while(l<r){
int mid=(l+r+1)/2;
if(a[mid]>=b[i]) r=mid-1;
else l=mid;
}
if(a[l]>=b[i]) l=-1;
int le=l+1;
l=0;
r=n-1;
while(l<r){
int mid=(l+r)/2;
if(b[i] >=c[mid]) l=mid+1;
else r=mid;
}
if(c[r]<=b[i]) r=n;
int lr=n-r;
cnt+=1ll*le*lr;
}
cout<<cnt<<endl;
}