본문 바로가기

회전하는 큐

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

 

1021번: 회전하는 큐

첫째 줄에 큐의 크기 N과 뽑아내려고 하는 수의 개수 M이 주어진다. N은 50보다 작거나 같은 자연수이고, M은 N보다 작거나 같은 자연수이다. 둘째 줄에는 지민이가 뽑아내려고 하는 수의 위치가

www.acmicpc.net

 

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
#include <iostream>
#include <deque>
#include <vector>
#include <algorithm>
 
using namespace std;
 
deque<int> a;
 
int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    int n, m;
    int left, right;
    int cnt = 0;
 
    cin >> n >> m;
 
    for (int i = 1; i <= n; i++)
        a.push_back(i);
 
    vector<int> x(m);
    for (int i = 0; i < m; i++)
        cin >> x[i];
 
    for (int i = 0; i < m; i++) {
        int num;
        num = x[i];
 
        for (int i = 0; i < a.size(); i++)
        {
            if (a[i] == num)
            {
                left = i;
                right = a.size() - i;
                break;
            }
        }
 
        if (left <= right) //왼쪽 밀기
        {
            while (1)
            {
                if (a.front() == num)   //제일 앞에 있는 경우
                    break;
                a.push_back(a.front());
                a.pop_front();
                cnt++;
            }
            a.pop_front();
        }
        else  // 오른쪽 밀기
        {
            cnt++// 정답도 밀어야하니까!!
            while (1)
            {
                if (a.back() == num)
                    break;
                a.push_front(a.back());
                a.pop_back();
                cnt++;
            }
            a.pop_back();
        }
    }
    
 
    cout << cnt;
    return 0;
}
cs

'' 카테고리의 다른 글

큐 회전  (0) 2024.10.30