🧩 Problem Solving/[백준]
[백준] 28250 이브, 프시케 그리고 푸른 MEX의 아내 (C++ cpp)
제봉아
2023. 7. 3. 15:44
https://www.acmicpc.net/problem/28250
28250번: 이브, 프시케 그리고 푸른 MEX의 아내
첫째 줄에 정수 $N$이 주어진다. ($2 \le N \le 200\,000$) 둘째 줄에 $N$개의 정수 $A_1, A_2, \dots, A_N$이 공백으로 구분되어 주어진다. ($0 \le A_i \le 100\,000$)
www.acmicpc.net
아이디어
https://aia1235.tistory.com/75
[백준] 28250 이브, 프시케 그리고 푸른 MEX의 아내 (python 파이썬)
https://www.acmicpc.net/problem/28250 28250번: 이브, 프시케 그리고 푸른 MEX의 아내 첫째 줄에 정수 $N$이 주어진다. ($2 \le N \le 200\,000$) 둘째 줄에 $N$개의 정수 $A_1, A_2, \dots, A_N$이 공백으로 구분되어 주어진
aia1235.tistory.com
전체 코드
#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;
}