题目链接:https://www.51nod.com/onlineJudge/questionCode.html#!problemId=1305
有这样一段程序,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对1是2,1对其他数是1,2对2是1,所以就是求1和2的个数
代码:
#include<bits/stdc++.h>
#define N 100009
using namespace std;
typedef long long LL;
const int inf = 0x3f3f3f3f;
int main()
{
int num, a, b, c, n, i;
scanf("%d", &n);
a = b = c = 0;
for(i = 1; i <= n; i++)
{
scanf("%d", &num);
if(num == 1) a++;
else if(num == 2) b++;
else c++;
}
int ans = a * (a - 1) + b * (b - 1) / 2 + a * (c + b);
cout << ans << endl;
return 0;
}