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

题目链接：http://www.51nod.com/onlineJudge/questionCode.html#!problemId=1305

分析：看大佬的分析，才知道，本人一直超时！不说了，统计1的个数和2的个数，有多少1就加一个1，有多少个2就是n*(n-1)/2！

下面给出AC代码：

 1 #include <bits/stdc++.h>
 2 using namespace std;
 3 int main()
 4 {
 5     int n;
 6     int a[100005];
 7     while(scanf("%d",&n)!=EOF)
 8     {
 9         for(int i=1;i<=n;i++)
10             scanf("%d",&a[i]);
11             sort(a+1,a+1+n);
12             int c1=0,c2=0,c3=0;
13             for(int i=1;i<=n;i++)
14             {
15                 if(a[i]==1)
16                  c1++;
17                  else if(a[i]==2)
18                         c2++;
19                  else c3++;
20             }
21             int ans=c1*(c1+c2+c3-1)+c2*(c2-1)/2;
22             printf("%d\n",ans);
23     }
24     return 0;
25 }