51Nod_1305 Pairwise Sum and Divide

                              51Nod_1305 Pairwise Sum and Divide

                                         http://www.51nod.com/Challenge/Problem.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。

输入

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

输出

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

样例输入

3
1 4 1

样例输出

4

分析

(A[i]+A[j])/(A[i]*A[j])实际为1/A[i]+1/A[j]，循环是让每两个数相加。由于要向下取整，因此可以统计1，2，和大于2的数的数量，分别用c1,c2和c记录。

情况一：两个元素均为1，每个组合的值为2，共有c1(c1-1)/2种组合

情况二：两个元素为1和大于1的数，由于要向下取整，因此每个组合的值为1，共有c1(c2+c)中组合

情况三：两个元素均为2，每个组合的值为1，共有c2(c2-1)/2种组合

情况四：两个元素均大于二，每个组合的值为0

因此结果为c1(c1-1)+c1(c2+c)+c2(c2-1)/2

C++程序

#include<iostream>

using namespace std;

int main()
{
	int n;
	long long c1=0,c2=0,c=0,ans;
	cin>>n;
	for(int i=0;i<n;i++){
		long long x;
		cin>>x;
		if(x==1)
		  c1++;
		else if(x==2)
		  c2++;
		else
		  c++;
	}
	ans=c1*(c1-1)+c1*(c2+c)+c2*(c2-1)/2;
	cout<<ans<<endl;
	return 0;
}