给定三个整数数组A=[A1,A2,…AN], B=[B1,B2,…BN],C=[C1,C2,…CN]
,请你统计有多少个三元组 (i,j,k)(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≤10^5
0≤Ai,Bi,Ci≤10^5
输入样例:3
1 1 1
2 2 2
3 3 3
输出样例:27
思路:N是100000,大概要设计一个nlogn的算法。很容易想到是直接O(n^3)的算法,遍历三个数组。这样肯定超时,思考的起点就是这个O(n),一定是遍历了某个数组,那么可以猜到是遍历了B数组,也就是对每一个B中的数,找出A中比他小的数的个数与C中大于他的个数的乘积之和。现在要解决的是如何找出A中的数的个数。
前缀和:
这个可以用前缀和解决。首先用cnt数组记录每个数出现的次数,cnt[i]表示A中i出现的次数。
在对他求前缀和s[]。那么B中的数B[j],比他小的个数就是s[B[j]-1].
#include <iostream>
#include <algorithm>
using namespace std;
const int N=1000020;
int A[N]={
0},B[N]={
0},C[N]={
0};
int cnt[N]={
0},s[N]={
0};
long long sa[N]={
0},sc[N]={
0};
int main()
{
int n;cin>>n;
for(int i=1;i<=n;i++)
cin>>A[i];
for(int i=1;i<=