【51nod】-1305 Pairwise Sum and Divide(数学）

1305 Pairwise Sum an d Divide

题目来源：  HackerRank

基准时间限制：1 秒 空间限制：131072 KB 分值: 5  难度：1级算法题

 收藏

 关注

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

题解：

求出1和2的个数

1与任何数做上诉运算结果都是1，都会在sum上加1，考虑其他数，这时不再考虑1，1已经和所有数运算过。

2只有与2做上述运算时结果才是1，其他都是0.

#include <cstdio>
#include <stack>
#include <queue>
#include <cmath>
#include <vector>
#include <cstring>
#include <algorithm>
using namespace std;
#define CLR(a,b) memset(a,b,sizeof(a))
#define INF 0x3f3f3f3f
#define LL long long
int main()
{
	int n;
	LL a;
	int cnt1=0;
	int cnt2=0;
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a);
		if(a==1)
			cnt1++;
		else if(a==2)
			cnt2++;
	}	
	LL sum=(LL)cnt1*(LL)(n-1)+((LL)cnt2*(LL)(cnt2-1))/2;
	printf("%lld\n",sum);
	return 0;
}