51nod 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。

 

输入

第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

思路

1、对于算式Floor((A[i]+A[j])/(A[i]*A[j])) 来说

 

• 如果其中A[i]、A[j]都是大于2的数值，那么对应的值一定是0；
• 如果（A[i]==2&&A[j]>2）||（A[i]>2&&A[j]==2），那么对应的值一定为0；
• 如果A[i]==1&&A[j]==1，那么对应的值一定为2；
• 如果（A[i]==1&&A[j]!=1）||（A[i]!=1&&A[j]==1），那么对应的值一定为1；
• 如果A[i]==2&&A[j]==2，那么对应的值一定为1.

因此要统计A[i]==1或者A[i]==2的个数。

2、对于每一个A[i]==1，其贡献出来的值为：n-1，如果有一个A[i]==1，那么对应结果应加上n-1.

接下来统计A[i]==2的个数，对应在最终结果上加count*（count-1）/2；

3、数据范围比较大， 注意使用long long.

 

#include<stdio.h>
int main()
{
    int n,a[100007],count=0;
    long long sum=0;
    scanf("%d",&n);
    for(int i=0;i<n;i++)
    {
        scanf("%d",&a[i]);
        if(a[i]==1)
        {
            sum+=n-1;
        }
        else if(a[i]==2)
        {
            count++;
        }
    }
    sum=sum+count*(count-1)/2;
    printf("%lld",sum);
    return 0;
}