[백준/C++] 1158번 요세푸스 문제 List iterator 사용

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

1158번: 요세푸스 문제

 

// 코드

#include <bits/stdc++.h>
using namespace std;

int main(){
    list<int> L;
    int N, K, e;
    cin >> N >> K;
    int num[N];
    
    for(int i = 1; i <= N; i++) L.push_back(i);
    list<int>::iterator t = L.begin();
    
    for(int j = 0; j < N; j++)
    {
        e = L.back(); 
        // list L의 끝수 저장
        for(int k = 0; k < K-1; k++)
        {
            	if(e == *t) t = L.begin(); 
                // *t가 끝수와 같을 경우 t에 첫수 주소
            	else t++; // 아니면 주소 +1
        }
        num[j] = *t; 
        // 삭제하는 순서대로 저장
        if(e == *t)
        {
            L.erase(t);
            t = L.begin(); 
            // 끝수를 지웠을 경우 t의 지점을 첫수 주소로 저장
        }
        else t = L.erase(t);
    }
    // 출력
    cout << "<";
    for(int k = 0; k < N; k++)
    	{
        	cout << num[k];
        	if(k != N-1) cout << ", ";
    	}
    cout << ">";
    return 0;
}