51nod1305 Pairwise Sum and Divide

題目鏈接：http://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)的計算結果。

Input示例

3
1 4 1

Output示例

4

AC代碼：

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<iostream>
#include<algorithm>
using namespace std;
int main()
{
    int a[100010],i,j,n;
    long long sum=0,x,y,z;
    cin>>n;
    x=y=z=0;
    for(i=0;i<n;i++)//只有1和2對結果起作用 
    {
        cin>>a[i];
        if(a[i]==1)
            x++;
        else if(a[i]==2)
            y++;
        else
            z++; 
    }
//  sort(a,a+n);//一開始的方法，最後一組數據出錯 
//  for(i=0;i<n;i++)
//      for(j=i+1;j<n;j++)
//      {
//          
//          x=(a[i]+a[j])/(a[i]*a[j]);
//          if(x==0) break;
//          sum+=x; 
//      }
    sum=x*(x+y+z-1)+y*(y-1)/2;
    cout<<sum<<endl;
return 0;
}