ZLOJ 三元组

题目描述
给定一个长度为N的数组A=[A1,A2,…AN]A=[A1,A2,…AN]，已知其中每个元素AiAi的值都只可能是1, 2或者3。
请求出有多少下标三元组(i, j, k)满足1 ≤ i < j < k ≤ N且Ai<Aj<AkAi<Aj<Ak。
输入
第一行包含一个整数N
第二行包含N个整数A1,A2,…ANA1,A2,…AN。(1≤Ai≤31≤Ai≤3)
输出
一个整数表示答案

样例输入
6
1 3 2 1 2 3

样例输出
3

提示
对于30%的数据，1 ≤ N ≤ 100
对于60%的数据，1 ≤ N ≤ 1000
对于100%的数据，1 ≤ N ≤ 100000
本题并不算太难，整体的思路是枚举所有等于2的元素的位置，再借助前缀和求出该位置前面等于1的元素的个数，再求出这个位置后面的等于3的元素的个数。唯一需要注意的地方就是数据类型。虽然题目的N只到十万，可是计算过程中会出现爆int的数据量，就……很shit。
上代码。

#include<iostream>
using namespace std;
long long n,temp,fi[100001],fk[100001],full[100001],aj[100001],j=1,ans;
int main(){
	cin>>n;
	for(int z=1;z<=n;z++){
		cin>>full[z];
		if(full[z]==2){
			aj[j]=z;
			j++;
		}
	}
	for(int z=1;z<=n;z++){
		fi[z]=fi[z-1];
		if(full[z]==1)
		    fi[z]++;
		if(full[z]==2)
		    temp++;
		fk[z]=z-fi[z]-temp;
	}
	for(int z=1;z<j;z++)
		ans+=fi[aj[z]]*(fk[n]-fk[aj[z]]);
	cout<<ans;
}