1021번 회전하는 큐

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

생각했던 조건들보다 1~2개 더 조건이 있었다.

deque로 구현하려고 하였으나, 큐 안에 있는 원소의 인덱스 위치를 찾기 귀찮아서

그냥 vector로 구현하였다.

vector의 insert와 erase를 알면 쉽게 풀 수 있다.

단, 오른쪽으로 이동할 때에는 tail이 하나 더 증가해야한다. (75번 째 줄의 erase 때문)

예)

1 2 3 4 5에서 4를 삭제할 경우, tail = |4-5| = 1이 되므로, 4 5 1 2 3 (총 2번 이동) 이 되어야 한다.

2를 삭제 할 경우에는 head = |1-2| = 1이 되므로, 2 3 4 5 1 (총 1번 이동)

#include <iostream>
#include <queue>
#include <stack>
#include <cstdio>
#include <vector>
#include <cstring>
#include <string>
#include <math.h>
#include <algorithm>
#include <map>
 
using namespace std;
 
vector<int> v, q;
 
int result = 0;
 
int main(void)
{
    int N = 0, M = 0, input = 0;
 
    scanf("%d %d", &N, &M);
 
    for (int i = 1; i <= N; i++)
    {
        q.push_back(i);
    }
 
    for (int i = 0; i < M; i++)
    {
        scanf("%d", &input);
 
        v.push_back(input);
    }
 
    for (int i = 0; i < v.size(); i++)
    {
        int idx = 1;
        int qSize = q.size();
 
        for (int j = 0; j < q.size(); j++)
        {
            if (v.at(i) == q.at(j)) break;
 
            ++idx;
        }
 
        int head = abs(1 - idx);
        int tail = abs(qSize - idx);
 
        if (head <= tail)
        {
            while (head--)
            {
                q.push_back(q.at(0));
                q.erase(q.begin());
 
                ++result;
            }
        }
        else
        {
            ++tail;
 
            while (tail--)
            {
                q.insert(q.begin(), q.at(q.size() - 1));
 
                q.erase(q.end() - 1);
 
                ++result;
            }
        }
 
        q.erase(q.begin());
    }
 
    printf("%d\n", result);
 
    return 0;
}