第九届蓝桥杯省赛C++ / JAVA B组《递增三元组》（c++）

1.题目说明

给定三个整数数组

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

2.输入格式

第一行包含一个整数 N。

第二行包含 N个整数 A1,A2,…AN。

第三行包含 N个整数 B1,B2,…BN。

第四行包含 N个整数 C1,C2,…CN。

3.输出格式

一个整数表示答案。

4.数据范围

1≤N≤10的5次方,
0≤Ai,Bi,Ci≤10的5次方

5.输入样例

3

1 1 1

2 2 2

3 3 3

6.输出样例

27

7.思路

本题的数据量大小为10的5次方，若用双层及以上多层循环必会超时。

我们可以考虑提前对A数组和C数组中数字出现次数进行预处理，即计算两个数组中从小到大数字出现次数的前缀和，这样可以直接得出比当前B[i]大或者小的数字有多少个。最后枚举B数组，将所有值加起来即为最终答案。

8.代码

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

using namespace std;

typedef long long LL;

const int N = 1e6 + 10;

int n;
int a[N], b[N], c[N];
LL as[N];  // as[i]表示在A[]中有多少个数小于b[i]
LL cs[N];  // cs[i]表示在C[]中有多少个数大于b[i]
int cnt[N], s[N];

int main()
{
    scanf("%d", &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] ++ ;

    // 求as[]
    for (int i = 0; i < n; i ++ ) cnt[a[i]] ++ ;
    for (int i = 1; i < N; i ++ ) s[i] = s[i - 1] + cnt[i];   // 求cnt[]的前缀和
    for (int i = 0; i < n; i ++ ) as[i] = s[b[i] - 1];

    // 求cs[]
    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]];

    // 枚举每个b[i]
    LL res = 0;
    for (int i = 0; i < n; i ++ ) res += as[i] * cs[i];

    cout << res << endl;

    return 0;
}