顺序三元组

来源:互联网 发布:云计算概念股60亿市值 编辑:程序博客网 时间:2024/05/29 04:30

给定一个长度为N的数组A=[A1, A2, ... AN],已知其中每个元素Ai的值都只可能是1, 2或者3。

请求出有多少下标三元组(i, j, k)满足1 ≤ i < j < k ≤ N且Ai < Aj < Ak

Input

第一行包含一个整数N  

第二行包含N个整数A1, A2, ... AN。(1 ≤ Ai ≤ 3)  

对于30%的数据,1 ≤ N ≤ 100  

对于80%的数据,1 ≤ N ≤ 1000  

对于100%的数据,1 ≤ N ≤ 100000  

Output

一个整数表示答案

Sample Input
61 3 2 1 2 3
Sample Output
3

思路:

先记录数组里3的个数T3;从前到后遍历寻找1的个数T1,

每当找到2,cnt+=T1*T3;(T1为2前边1的个数,T3为2后边3的个数)


代码:


#include<stdio.h>#define MAX 100005using namespace std;typedef long long ll;ll arr[MAX];ll t1,t3;int main(){    ll n;    scanf("%lld",&n);    ll i;    for(i=0;i<n;i++)        scanf("%lld",&arr[i]),t3+=arr[i]==3;    ll cnt=0;    for(i=0;i<n;i++){        t1+=arr[i]==1;        t3-=arr[i]==3;        cnt+=(arr[i]==2)*(t1*t3);    }    printf("%lld\n",cnt);return 0;}


原创粉丝点击