취미로 코딩하기

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

이번에 볼 문제는 백준 26738번 문제인 Lizak이다.
문제는 아래 링크를 확인하자.

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

각 수열의 원소별로 해당 원소를 담아 보관하는 큐 자료구조를 대응시켜 시뮬레이션으로 문제를 해결하자.

각 원소와 큐를 대응시키는 것은 map을 이용해 간단히 할 수 있다.

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

#include <iostream>
#include <queue>
#include <map>
using namespace std;

int N;
map<int, queue<int>> mp;

int ans = 1000000007;

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

	cin >> N;
	for (int i = 0; i < N; i++) {
		int x; cin >> x;
		if (mp.count(x) == 0) mp.insert({ x,queue<int>() });
		auto& que = mp[x];
		que.push(i);
		if (que.size() == 3) {
			ans = min(ans, que.back() - que.front() + 1);
			que.pop();
		}
	}

	if (ans == 1000000007) cout << "NIE";
	else cout << ans;
}