51Nod 1305 Pairwise Sum and Divide

这个水题很有意思，明显我们发现所求的值，只和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;
}