【ACWing】1236. 递增三元组

题目地址：

https://www.acwing.com/problem/content/description/1238/

给定三个整数数组：$$\begin{gathered} A=[A_1,A_2,\dots A_N] \\ B=[B_1,B_2,\dots B_N] \\ C=[C_1,C_2,\dots C_N] \end{gathered}$$请你统计有多少个三元组$(i,j,k)$满足：$1\le i,j,k\le N\wedge A_i<B_j<C_k$。

输入格式：
第一行包含一个整数$N$。
第二行包含$N$个整数$A_1,A_2,\dots A_N$。
第三行包含$N$个整数$B_1,B_2,\dots B_N$。
第四行包含$N$个整数$C_1,C_2,\dots C_N$。

输出格式：
一个整数表示答案。

数据范围：
$1\le N\le 10^5$
$0\le A_i,B_i,C_i\le 10^5$

枚举$B_j$，对于每一个$B_j$，我们需要知道在$A$里有多少个数小于$B_j$，和在$C$里有多少个数大于$B_j$。求比$B_j$小的数有多少个，可以通过对计数数组求前缀和来做。先对$A$求计数数组$a$（即$a[k]$表示$k$在$A$中出现了多少次），然后对$a$求前缀和，这样$a[k]$的含义就变为$A$中小于等于$k$的数有多少个。对$C$也可以类似操作，求计数数组然后求后缀和。代码如下：

#include <iostream>
using namespace std;
using ll = long long;

const int N = 1e5 + 10;
int n;
int b[N];
int pa[N], pb[N], pc[N];

int main() {
  scanf("%d", &n);
  for (int i = 1, x; i <= n; i++) {
    scanf("%d", &x);
    pa[x]++;
  }
  for (int i = 1, x; i <= n; i++) scanf("%d", &b[i]);
  for (int i = 1, x; i <= n; i++) {
    scanf("%d", &x);
    pb[x]++;
  }

  for (int i = 1; i < N - 2; i++) pa[i] += pa[i - 1];
  for (int i = N - 2; i >= 0; i--) pb[i] += pb[i + 1];

  ll res = 0;
  for (int i = 1; i <= n; i++) {
    ll l = b[i] ? pa[b[i] - 1] : 0, r = pb[b[i] + 1];
    res += l * r;
  }

  printf("%lld\n", res);
}

时间复杂度$O(N)$，空间$O(R)$，$R$为所有数的值域。