BOJ 1021 : 회전하는 큐

• 문제 링크 : boj.kr/1021
• 난이도 : S3
• 태그 : 덱

1021번: 회전하는 큐

---

코드

#include <bits/stdc++.h>

#define all(x) (x).begin(), (x).end()

#define INF 0x7FFFFFFF

using namespace std;

using ll = long long;
using ld = long double;
using pii = pair<int,int>;
using pll = pair<ll, ll>;

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

    int n, m;
    cin >> n >> m;
    deque<int> _dq;
    for(int i = 1; i <= n; i++) {
        _dq.emplace_back(i);
    }
    vector<int> arr(m);
    for(auto &u : arr) {
        cin >> u;
    }
    int ans = 0;
    for(auto u : arr) {
        int i;
        for(i = 0; i < _dq.size(); i++) {
            if(_dq[i] == u) break;
        }
        if(i < _dq.size() - i) {
            for(int j = 0; j < i; j++) {
                _dq.push_back(_dq[0]);
                _dq.pop_front();
                ans++;
            }
            _dq.pop_front();
        }
        else {
            for(int j = 0; j < _dq.size() - i; j++) {
                _dq.push_front(_dq[_dq.size()-1]);
                _dq.pop_back();
                ans++;
            }
            _dq.pop_front();
        }
    }
    cout << ans;

    return 0;
}

---

풀이

큐가 회전한다는 것은 한 쪽에서 뺴서 다른 한 쪽으로 넣는 것과 같다.

덱을 통해 원소를 관리할 수 있다.

어떤 한 원소를 빼려고 할 때, 2번 연산과 3번 연산을 섞어서 사용할 필요가 없다.

둘 중 한 연산만 사용했을 때 더 적은 연산을 사용하는 쪽으로 연산을 수행하면 된다.