题目描述
给定三个整数数组
A=[A1,A2,…AN],
B=[B1,B2,…BN],
C=[C1,C2,…CN],
请你统计有多少个三元组 (i,j,k) 满足:
1.1≤i,j,k≤N
2.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
算法(整数二分)
用时:564ms
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=100010;
int n;
int a[N],b[N],c[N];
int www(int d[],int u)//大于u的元素个数
{
int l=0,r=n-1;
while(l<r){
int mid=l+r+1>>1;
if(d[mid]<=u) l=mid;
else r=mid-1;
}
if(d[l]<=u) return n-l-1;
else return n-l;
}
int qqq(int d[],int u)//小于u的元素个数
{
int l=0,r=n-1;
while(l<r){
int mid=l+r>>1;
if(d[mid]>=u) r=mid;
else l=mid+1;
}
if(d[l]>=u) return l;
else return l+1;
}
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(c,c+n);
long long int res=0;
for(int i=0;i<n;i++){
res+=1ll*qqq(a,b[i])*www(c,b[i]);
}
cout<<res<<endl;
return 0;
}