51Nod-1305 Pairwise Sum and Divide

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)的计算结果。
Input示例
3
1 

4

1
Output示例
4

题意：给定一个函数求函数值。

题解：可以得出 Floor((A[i]+A[j])/(A[i]*A[j])) 不为零的情况有A[i]=2，A[j]=2此时得1，A[i]=1,A[j]!=1此时得1，A[i]=1,A[j]=1此时得2.

    所以可以求得数组中1和2的个数使用组合数求解。

AC代码：

#include<iostream>
#include<string>
using namespace std;
int main()
{
	int n;
	int s[100007];
	cin>>n;
	for(int i=0;i<n;i++)cin>>s[i];
	int a=0;
	int b=0;
	for(int i=0;i<n;i++)
	{
		if(s[i]==1)a++;
		else if(s[i]==2)b++;
	}
	long long sum = a*(n-a)+a*(a-1)+b*(b-1)/2;
	cout<<sum<<endl;
	return 0;
}