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

https://www.51nod.com/Challenge/Problem.html#!#problemId=1305 

解题思路

直接按照给出的公式计算的话，一定会超时的，所以，需要找到它的规律。将它的公式化简过后，等于1/a[i]+1/a[j]，结果一共有三种情况，第一种是a[i]=1,a[j]=1，结果为2；第二种是a[i]=1,a[j]=任意数(除了1)，结果为1；第三种是a[i]=2,a[j]=2，结果为1。其他情况都为0，求和时不用计入。所以将问题转换为求1，2的个数。

代码

#include<stdio.h>

int main()
{
	int n,i,k;
	int sum=0,x1=0,x2=0,x3=0;
	scanf("%d",&n);
	for(i=0;i<n;i++)       //求出1，2和其他数的个数 
	{
		scanf("%d",&k);	
		if(k==1)
		{
			x1++;
		}
		else if(k==2)
		{
			x2++;
		}
		else
		{
			x3++;
		}
	}	
	sum=(x1*(x1-1)+x1*(x3+x2)+(x2*(x2-1)/2));//将三种情况相加 
	printf("%d",sum);
	return 0;
}