这个水题很有意思,明显我们发现所求的值,只和1,2的存在有关
统计一下 推出公式就行Orz
#include <bits/stdc++.h>
using namespace std;
#define ls st<<1
#define rs st<<1|1
#define LL long long
const int MAXN= (int) 1e6+10;
int arr[MAXN];
int main() {
ios::sync_with_stdio(false);
int n, x;
cin >> n;
int x1=0, x2=0, x3=0;
for (int i = 0; i < n; ++i) {
cin >> x;
if (x == 1)
x1++;
else if (x == 2)
x2++;
else
x3++;
}
cout << x1*(x1+x2+x3-1)+x2*(x2-1)/2 << endl;
return 0;
}