前缀和O(n)时间复杂度
a[i]表示a中大小为i的个数,c[i]表示c中大小为i的个数,b[i]依次存入
hash表求出a[i]中比b[i]小的个数和c[i]中比b[i]大的个数
两个数相乘,将结果依次加入ans求出结果。
#include <iostream>
using namespace std;
const int N=1e5+10;
int a[N],b[N],c[N];
int n;
long long int ans;
int main() {
cin >> n;
int num;
//hash存入a[i];
for(int i=0; i<n; i++) {
cin >> num;
a[++num]++;//防止b[i]==0情况出现,见代码尾
}
for(int i=1; i<N-10; i++)
a[i+1]+=a[i];
//b[i]直接存入后续查找a[i],c[i]所需要的数目
for(int i=0; i<n; i++)
cin >> b[i];
//c[i]同样hash存入
for(int i=0; i<n; i++) {
cin >> num;
c[num]++;
}
for(int i=0; i<N-10; i++)
c[i+1]+=c[i];
for(int i=0; i<n; i++)
ans+=(long long int)a[b[i]]*(n-c[b[i]]);//前面如果使用a[num]++,此处需要用a[b[i]-1],b[i]=0时越界
cout << ans;
return 0;
}