51NOD 1305 Pairwise Sum and Divide

题目链接：https://www.51nod.com/onlineJudge/questionCode.html#!problemId=1305

1305 Pairwise Sum and 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示例

31 4 1

Output示例

4

解析：这个直接暴力双重循环，必然超时，1对1是2,1对其他数是1,2对2是1，所以就是求1和2的个数

代码：

#include<bits/stdc++.h>#define N 100009using namespace std;typedef long long LL;const int inf = 0x3f3f3f3f;int main(){    int num, a, b, c, n, i;    scanf("%d", &n);    a = b = c = 0;    for(i = 1; i <= n; i++)    {        scanf("%d", &num);        if(num == 1) a++;        else if(num == 2) b++;        else c++;    }    int ans = a * (a - 1) + b * (b - 1) / 2 + a * (c + b);    cout << ans << endl;    return 0;}