1305 Pairwise Sum and Divide(排列组合）

有这样一段程序，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。
输入
第1行：1个数N，表示数组A的长度(1 <= N <= 100000)。
第2 - N + 1行：每行1个数A[i]（1 <= A[i] <= 10^9)。
输出
输出fun(A)的计算结果。
输入样例

3
1 4 1
输出样例
4
（1，1）： 2
(x-1)+(x-2)+…+1=x(x-1)/2
（1，>1): 1
x*y
(2,2): 1
（x-1）+（x-2)+…+1=x(x-1)/2

#include<bits/stdc++.h>
using namespace std;
int main()
{
  ios::sync_with_stdio(false);
  int n,sum=0,one=0,two=0,other=0,tmp;
  cin>>n;
  for(int i=1;i<=n;i++)
  {
  	cin>>tmp;
  	if(tmp==1) one++;
  	else if(tmp==2) two++;
  	else other++;
  }
  cout<<one*(one-1)+one*(two+other)+two*(two-1)/2; 
  return 0;
}