AcWing 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≤105,
0≤Ai,Bi,Ci≤105
输入样例:
3
1 1 1
2 2 2
3 3 3
输出样例:
27
#include<bits/stdc++.h>
using namespace std;
const int N = 100010;
typedef long long LL;
int a[N],b[N],c[N];
int as[N];//表示有多少A小于B
int cs[N];//表示有多少B小于C
int cnt[N],s[N];
int n;
int main()
{
cin>>n;
//输入每一行的数字,共n行,并计算每行有多少个数字
for(int i = 0;i<n;i++) cin>>a[i],a[i]++;
for(int i = 0;i<n;i++) cin>>b[i],b[i]++;
for(int i = 0;i<n;i++) cin>>c[i],c[i]++;
//求as[]
for(int i = 0;i<n;i++) cnt[a[i]]++;//计算对应每个a[i]的数量
for(int i = 1;i<N;i++) s[i] = s[i-1]+cnt[i];//求cnt[N]的前缀和
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]]++;//计算对应每个c[i]的数量
for(int i = 1;i<N;i++)s[i] = s[i-1]+cnt[i];//求cnt[N]的前缀和
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 +=(LL)as[i]*cs[i];//求出有多少个三元组合, 例如题干中给的结果3*3*3
cout<<res<<endl;
return 0;
}