【蓝桥杯】递增三元组

给定三个整数数组

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≤1e5
0≤Ai,Bi,Ci≤1e5

输入样例：

3
1 1 1
2 2 2
3 3 3

输出样例：

27

思路：

        从题目出发， Ai < Bj < Ck，三层for循环遍历一定超时，容易想到处理A，B，C的前缀和通过乘法来计算一共有多少个符合条件的三元组，从A,C两头处理这个问题都不太容易，而从B出发分别计算比 Bi 小的 A 数组中有多少个数和 比 Bi 大的C数组有多少个数更合适。

思路实现：

1.计算比 Bi 小的 A 数组中有多少个数

    for(int i = 0; i < n; i++)
        cnt[a[i]]++;
    for(int i = 1; i < N; i++)
        s[i] = s[i - 1] + cnt[i];
    for(int i = 0; i < n; i++)
        as[i] = s[b[i] - 1];

 2.计算比 Bi 大的 C 数组有多少个数

    for(int i = 0; i < n; i++)
        cnt[c[i]]++;
    for(int i = 1; i < N; i++)
        s[i] = s[i - 1] + cnt[i];
    for(int i = 0; i < n; i++)
        cs[i] = s[N - 1] - s[b[i]];

完整代码（C++）:

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>

using namespace std;

const int N = 1e5 + 10;
int a[N], b[N], c[N];
int as[N], cs[N];
int cnt[N], s[N];

int main()
{
    int n;
    cin >> n;
    
    for(int i = 0; i < n; i++)
        scanf("%d", &a[i]), a[i]++;
    for(int i = 0; i < n; i++)
        scanf("%d", &b[i]), b[i]++;
    for(int i = 0; i < n; i++)
        scanf("%d", &c[i]), c[i]++;
        
    for(int i = 0; i < n; i++)
        cnt[a[i]]++;
    for(int i = 1; i < N; i++)
        s[i] = s[i - 1] + cnt[i];
    for(int i = 0; i < n; i++)
        as[i] = s[b[i] - 1];
        
    memset(cnt, 0, sizeof cnt);
    memset(s, 0, sizeof s);
    
    for(int i = 0; i < n; i++)
        cnt[c[i]]++;
    for(int i = 1; i < N; i++)
        s[i] = s[i - 1] + cnt[i];
    for(int i = 0; i < n; i++)
        cs[i] = s[N - 1] - s[b[i]];
        
    long long res = 0;
    for(int i = 0; i < n; i++)
        //计算之前需要强转为long long，as与cs的乘法有可能爆int
        res += (long long) as[i] * cs[i]; 
        
    cout << res << endl;
    
    return 0;
}