有这样一段程序,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;
}