algorithmStudy/11866.cpp at master · judy5050/algorithmStudy · GitHub

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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
//
//  11866.cpp
//  algo
//
//  Created by 박효정 on 2021/10/20.
//

#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>

using namespace std;

int main(){

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


    queue<int>q1;
    queue<int>q2;
    queue<int>res;
    int n,k;
    cin>>n>>k;
    for(int i=1;i<=n;i++){
        q1.push(i);
    }

    while(!q1.empty()){

        for(int i=0;i<k-1;i++){
            if(!q1.empty()){
                q2.push(q1.front());
                q1.pop();
            }else{
                while(!q2.empty()){
                    q1.push(q2.front());
                    q2.pop();

                }
                i--;
            }

        }

        if(!q1.empty()){
            res.push(q1.front());
            q1.pop();
        }else{
            res.push(q2.front());
            q2.pop();
        }


        while(!q2.empty()){
            q1.push(q2.front());
            q2.pop();

        }


    }

    cout<<'<';
    while(!res.empty()){

        if(res.size()>1){
            cout<<res.front()<<", ";
            res.pop();
        }else{
            cout<<res.front()<<'>'<<"\n";
            res.pop();
        }


    }


    return 0;
}