1236. 递增三元组（枚举、桶排序、前缀和）——第九届蓝桥杯省赛B组

1236. 递增三元组

给定三个整数数组

A=[A1,A2,…AN],
B=[B1,B2,…BN],
C=[C1,C2,…CN],

请你统计有多少个三元组 (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

题解：
求三元组的数量，并且A<B<C
可以遍历B数组，每次将小于B的A数组中的个数和大于B的C数组中的个数乘起来，并累加就是要求的答案。
因为数据范围是10^5，因此我们需要一个O(n)或者O(nlogn)的算法才能通过此题
10⁵的范围不算大，我们可以开一个这样大的数组，使用桶排序得到某个数组中每个数出现次数，再使用前缀和得到小于等于每个数的个数，这样利用前缀和可以轻松求得某个范围内数的个数，因此利用前缀和求小于B的A数组中的个数、大于B的C数组中的个数这样的值是轻而易举的
在代码中我们把每次遍历B数组获得的小于B的A数组中的个数、大于B的C数组中的个数存在专门的数组as[N]和cs[N]里面，最后使用

//求as[i]*cs[i]
    LL res = 0;
    for (int i = 0; i < n; i ++ ){
        LL t = (LL)as[i]*cs[i];
        res += t;
    }

代码即可求解

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 100005;
typedef long long LL;
int cnt[N],s[N],a[N],b[N],c[N];
int as[N],cs[N];

int main()
{
    int n;
    cin>>n;
    //这里记得将输入的数全体加1，避免0的出现，干扰后面前缀和数组的计算
    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]]++;
    }
    //因为这里前缀和数组下标从1开始，加不到cnt[0], 所以不能有数值为0的数
    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];
    }
    
    //求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]];
    }
    
    //求as[i]*cs[i]
    LL res = 0;
    for (int i = 0; i < n; i ++ ){
        LL t = (LL)as[i]*cs[i];
        res += t;
    }
    printf("%lld\n",res);
    return 0;
}