有这样一段程序,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时结果才可能不为0,其余情况(A[i]+A[j])/(A[i]*A[j]))均为0;
1:(1+1)/(1*1)=2.
2:(1+x)/(1*x)=1(x>1)
3;:(2+2)/(2*2)=1;
其余情况为0;
#include<stdio.h>
int a[100200];
int n,s,t;
int main()
{
while(~scanf("%d",&n))
{
s=0,t=0;
for(int i=0;i<n;i++)
{
scanf("%d",&a[i]);
if(a[i]==1)s++;
if(a[i]==2)t++;
}
int sum=0;
sum+=(n-s)*s;//判 2;
sum+=(s-1)*s;//判 1;
sum+=t*(t-1)/2; //判3;
printf("%d\n",sum);
}
return 0;
}