51nod 1305 Pairwise Sum and Divide

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

题目：

有这样一段程序，fun会对整数数组A进行求值，其中Floor表示向下取整：

fun(A)

    sum = 0

    for i = 1 to A.length

        for j = i+1 to A.length

            sum = sum + Floor((A[i]+A[j])/(A[i]*A[j])) 

    return sum

给出数组A，由你来计算fun(A)的结果。例如：A = {1, 4, 1}，fun(A) = [5/4] + [2/1] + [5/4] = 1 + 2 + 1 = 4。

Input

第1行：1个数N，表示数组A的长度(1 <= N <= 100000)。
第2 - N + 1行：每行1个数A[i]（1 <= A[i] <= 10^9)。

Output

输出fun(A)的计算结果。

只需要考虑1和2。对1内部组合，每个组合值为2。1和别的数组合，每个组合值为1。对2内部组合，每个组合值为1。

#include <iostream>
#include<bits/stdc++.h>

using namespace std;

int main()
{
    int n;
    scanf("%d",&n);
    int one=0,two=0;
    for(int i=0;i<n;i++)
    {
        int x;
        scanf("%d",&x);
        if(x==1)
            one++;
        if(x==2)
            two++;
    }
    int ans=two*(two-1)/2+(one-1)*one+one*(n-one);
    cout<<ans<<endl;
}