51nod1305(简单逻辑)

题目链接：https://www.51nod.com/onlineJudge/questionCode.html#!problemId=1305

题意：中文题诶～

思路：1e5的数据直接暴力肯定是不行的，我们不难想到：

(a+b)/(a*b)的结果只有４种情况：

１．a=b=1, 结果为２；

2.   a=1, b!=1,　结果为１；

３.  a==b==2，　结果为１;

4.　其他情况，　结果为０；

 

代码：

#include <bits/stdc++.h>
#define MAXN 100010
using namespace std;

int a[MAXN], gg1[MAXN], gg2[MAXN];

int main(void){
    int n, ans=0;
    cin >> n;
    for(int i=0; i<n; i++){
        cin >> a[i];
    }
    for(int i=n-1; i>=0; i--){
        gg1[i]=gg1[i+1]+(a[i]==1);  //gg1[i]存储i后面１的数目(包括第i个数)
        gg2[i]=gg2[i+1]+(a[i]==2);  //gg1[i]存储i后面2的数目(包括第i个数)
    }
    for(int i=0; i<n; i++){
        if(a[i]==1){
            ans+=gg1[i+1]+(n-i-1);
        }else if(a[i]==2){
            ans+=gg1[i+1]+gg2[i+1];
        }else{
            ans+=gg1[i+1];
        }
    }
    cout << ans << endl;
}