취미로 코딩하기

※ 글쓴이는 취미로 코딩을 익혀보는 사람이라 정확하지 않은 내용을 담고 있을 수 있다 ※

이번에 볼 문제는 백준 1337번 문제인 올바른 배열이다.
문제는 아래 링크를 확인하자.

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

주어진 수로 시작하지 않는 다섯개의 쌍을 체크할 수 있는 경우 다섯 수 중 가장 작은 수를 제외하고 가장 큰 수의 다음 수를 추가하는 것을 반복하는 것으로 같은(또는 더 적은) 개수의 수를 추가해 다섯 개의 연속한 수가 들어있게 만들 수 있음을 관찰하자.

따라서 주어지는 각 수 x에 대하여 x+1, x+2, x+3, x+4 중 주어진 배열에 존재하지 않는 수의 개수를 체크해나가는 것으로 문제를 해결할 수 있다. 이는 set을 이용해 쉽게 구현할 수 있다.

아래는 제출한 소스코드이다.

#include <iostream>
#include <set>
using namespace std;

int N;
set<int> st;

int ans = 5;

int main() {
	ios::sync_with_stdio(0);
	cin.tie(0);

	cin >> N;
	while (N--) {
		int x; cin >> x;
		st.insert(x);
	}

	for (auto& x : st) {
		int cnt = 4;
		for (int i = 1; i < 5; i++) {
			if (st.count(x + i)) cnt--;
		}
		ans = min(ans, cnt);
	}

	cout << ans;
}