AcWing 1236. 递增三元组(枚举 + 前缀和）

题目

思路：
如果直接暴力 三个 for, O(n^3) 超时，所以优化。
关键是b 它是中间值，找出a []中比它小的数的个数as，找出b[]中比它大的数的个数cs，as * cs即为答案

分析图：

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;

const int N=100010;
typedef long long LL;

int n,a[N],b[N],c[N];
int as[N];//as[N] 表示 A[] 中有多少个小于 b[i]
int cs[N];//cs[N] 表示 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]++;//都加 1 是因为：前缀和计算时，数组下标从 1 开始 
	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]]++;//再用前缀和时，这也是另一种排序，这个只需要 O(n) ，而其他的排序一般都是 O(n^2) 
	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]; // 求cnt[]的前缀和
	for(int i=0;i<n;i++) cs[i]=s[N-1]-s[b[i]];
	 
	LL res=0;
	for(int i=0;i<n;i++) res+=(LL)as[i]*cs[i];//因为得出的答案可能会爆 int 所以加上 LL, 强制类型转换 
	 
	 cout<<res<<endl;
	 
	return 0;
}