[백준] 28250 이브, 프시케 그리고 푸른 MEX의 아내 (C++ cpp)

https://www.acmicpc.net/problem/28250

28250번: 이브, 프시케 그리고 푸른 MEX의 아내

---

아이디어

https://aia1235.tistory.com/75

[백준] 28250 이브, 프시케 그리고 푸른 MEX의 아내 (python 파이썬)

---

전체 코드

#include<iostream>
using namespace std;

int N = 0, num = 0;
long count_0 = 0, count_1 = 0;

int main()
{   
    cin.tie(0)->sync_with_stdio(0);
    cin >> N;

    for (int i = 0; i < N; i++){
		cin >> num;
        if(num == 0)
            count_0 = count_0 + 1;
        else if(num == 1)
            count_1 = count_1 + 1;
    }

    cout << count_0 * count_1 * 2 + count_0 * (N - count_0 - count_1) + ((count_0 * (count_0 - 1)) / 2) << endl;
    
    return 0;
}