51nod 1305Pairwise Sum and Divide

最新推荐文章于 2018-09-06 02:02:51 发布

linmory

最新推荐文章于 2018-09-06 02:02:51 发布

阅读量108

点赞数

分类专栏：数论

本文链接：https://blog.csdn.net/lpeaceminusone/article/details/81136948

版权

数论专栏收录该内容

39 篇文章 0 订阅

订阅专栏

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示例

3
1 4 1

Output示例

求(a[i]+a[j])/a[i]*a[j]就是求(1/a[i]+1/a[j]),每个a[i]出现次数是n-1次。

又因为需要向下取整所以，结果只跟1和2出现的次数有关，当a[i]或a[j]==1时，或者a[i]和a[j]==2时，所以求出1和2出现次数就可以了。

#include<map>
#include<stack>
#include<queue>
#include<math.h>
#include<vector>
#include<string>
#include<stdio.h>
#include<iostream>
#include<string.h>
#include<algorithm>
#define maxn 1005
#define inf 0x3f3f3f3f3f
#define ll  long long
#define mod 10
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;
int main(){
   int n;
   while(cin>>n){
        ll x,ans=0,a1=0,a2=0;
        for(int i=0;i<n;i++){
            scanf("%lld",&x);
            if(x==1)a1++;
            if(x==2)a2++;
        }
        ans=(n-1)*a1+(a2*(a2-1)*0.5);
        cout<<ans<<endl;
   }
}