Little Jay

[C++] 백준 1021번 회전하는 큐 본문

알고리즘/BOJ

[C++] 백준 1021번 회전하는 큐

Jay, Lee 2021. 8. 4. 12:58

제목 그대로 회전하는 큐이다.

회전하는 큐의 의미를 잘 생각해본다면 자료구조 중 하나인 데큐를 사용하면 편한 문제이다.

푸는 방법은 

deque 의 사이즈를 통해서 찾는 것이 효율적이다.
temp_index로 숫자를 찾고 size의 반보다 cin한 숫자의 index가 크면
뒤로 pop_back() & push_front()
아니면 pop_front() & push_back()을 하면 된다

 

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

int main() {

	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);

	int n, m, x;
	cin >> n >> m;

	int count = 0;

	deque<int> dq;

	for (int i = 0; i < n; i++) {
		dq.push_back(i + 1);
	}

	for (int i = 0; i < m; i++) {
		cin >> x;
		int temp_index = 0;

		for (int k = 0; k < dq.size(); k++) {
			if (dq[temp_index] == x)
				break;
			temp_index++;
		}

		if (temp_index > dq.size() / 2) {
			while (dq.front() != x) {
				int temp_int = dq.back();
				dq.pop_back();
				dq.push_front(temp_int);
				count++;
			}
		}
		else {
			while (dq.front() != x) {
				int temp_int = dq.front();
				dq.pop_front();
				dq.push_back(temp_int);
				count++;
			}
		}
		dq.pop_front();
	}
	cout << count << "\n";

	return 0;
}
Comments