1305 Pairwise Sum and Divide

有这样一段程序，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

#include<iostream>
#include<string>
using namespace std;
long long int sum;
int n;
int a[100000];
int sum_1 = 0; 
int sum_2 = 0;
int res = 0;
int main()
{
	int i;
	int j;
	cin >> n;
	for(i = 0;i < n;i++)
	{
		cin >> a[i];
		if(a[i] == 1)
		{
			sum_1++;
		}
		if(a[i] == 2)
		{
			sum_2++;
		}
	}
	res += (sum_1*(sum_1 - 1))/2;
	res += (sum_2*(sum_2 - 1))/2;
	
	for(i = 0;i < sum_1;i++)
	{
		res += (n - i -1);
	 } 
	
	
	cout << res; }