5-H0ngJu

[H0ngJu]

🔗 문제 링크

요세푸스 문제

✔️ 소요된 시간

40분

✨ 수도 코드

• location : 사람 위치
• cnt : index 기준 k번째 카운트

1. table에 앉은 사람이 없을 때까지 아래를 반복한다.
   1. k번째 사람인지 확인하기 위해 cnt를 증가시키며 k-1(인덱스기준)인지 확인한다.
      1. k번째 사람이라면, table에서 pop하고 answer에 넣는다.
      2. 사람을 pop하였으므로 location은 - 1
   2. k번째 사람이 아니라면, cnt를 증가시킨다.

처음에 문제를 보고 연결리스트인가? 했지만, 리스트로도 풀 수 있을 것 같아서 그냥 리스트로 접근했다.
그런데 문제를 풀고, 다른 사람들의 풀이를 보니 '큐'를 사용하면 더 간단하게 풀린다는 것을 뒤늦게 깨달았다.

큐를 떠올리지 못했던 이유는 큐가 선입선출인 특징이 있는 것에 반해, 이 문제는 먼저 들어온 것이 먼저 나가는가?에 대해 아니라고 판단했기 때문이다.

그런데, 조금만 생각해보면 index 기준으로 k번째가 아니면, pop하고 다시 push함으로서 충분히 큐로도 풀 수 있는 문제였다. 😅😅

📚 새롭게 알게된 내용

• 리스트 출력 방법
  리스트를 문자열로 변환해서 출력하는 경우, 개별 요소를 각각 출력하는 것 보다, join()을 사용하는 것이 더 효율적이다. = 내장함수가 더 빠르다

처음에는 문제에서 요구하는 출력 포맷을 맞추기 위해서 아래와 같이 작성했다.

for i in range(len(solution(n, k))):
    if i == 0:
        print("<", end="")
        print(answer[i], end=", ")
    elif i == len(answer) - 1:
        print(answer[i], end="")
        print(">")
    else: print(answer[i], end=", ")

그런데 계속 시간 초과가 나서, 출력하는 부분을 아래와 같이 수정하였더니 사진과 같은 실행 시간이 나왔고, 통과가 되었다.

print("<" + ", ".join(map(str, answer)) + ">")

[tgyuuAn]

import sys

def input(): return sys.stdin.readline().rstrip()

N, K = map(int,input().split())

remain = [ x for x in range(1,N+1) ]
answer = []

index = 0
while remain:
    n = len(remain)
    index += K
    index = index % n 
    answer.append(remain[index-1])
    remain.pop(index-1)
    if index != 0: index -= 1
    
print("<", end="")
print(*answer, end="", sep=", ")
print(">", end="")

이거 그냥 어렵게 풀지않아도 매번 리스트에 있는거 pop(n)으로 뽑아도 $O(n^2)$이니까 풀리겠는데? 하고 바로 슛 해보니 맞네요!

뭔가 이 문제 크기를 늘리면 더 어려운 문제가 될 것 같은 느낌이 팍팍 드네요..

고생하셨씁니다.

다음 번엔 구현 말고 자료 구조 쪽으로 가봅시다!!!

[tgyuuAn]

제가 쓴 방법 보다 더 간결하고 좋네요 👍👍👍

[pknujsp]

이 코드 길이는 차이가 없는데 range로 간편하게 만들어 줄수있어요

table = list(range(1, n + 1))

[H0ngJu]

오오 그러네요 👏👏

[pknujsp]

인덱스를 K만큼 계산하고
리스트에서 뺸 후 요세푸스 순열 리스트에 추가 하는 걸

반복하는 방식으로 복잡하게 생각했는데

천천히 고민해보니 큐를 가지고
K - 1 개를 pop()해서 다시 뒤에 추가하고
맨 앞에 있는 원소를 요세푸스 순열에 추가하는 식으로 진행하면 되는 간단한 문제였군요.

N, K = map(int, input().split())
q = deque(range(1, N + 1))

result = []
while q:
    for _ in range(K - 1):
        q.append(q.popleft())
    result.append(q.popleft())

print(f"{'<'}{', '.join(map(str, result))}{'>'}")

N, K의 크기가 엄청 크면 분명히 통과가 안되겠다 싶어서
찾아보니 역시나 조건이 5000에서 100,000까지 상향된 조건으로
플래티넘3 짜리인 요세푸스 2문제가 있네요.

[Munbin-Lee]

#include <iostream>
#include <vector>
#include <numeric>

using namespace std;

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);

    int N, K;
    cin >> N >> K;

    vector<int> v(N);
    iota(v.begin(), v.end(), 1);

    int index = K - 1;

    cout << '<';

    while (--N) {
        cout << v[index] << ", ";
        v.erase(v.begin() + index);

        index += K - 1;
        index %= (int) v.size();
    }

    cout << v[0] << '>';

    return 0;
}

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

플래티넘 요세푸스 문제 풀다가 GG쳤습니다...

성장해서 언젠가는 PR 올리겠습니다.

[Munbin-Lee]

N, K = map(int, input().split())

people = list(range(1, N + 1))
index = 0

while people:
    index += K - 1
    index %= len(people)
    print(people.pop(index))

이런 식의 풀이는 어떨까요?

[H0ngJu]

N, K = map(int, input().split())

people = list(range(1, N + 1))
index = 0

while people:
    index += K - 1
    index %= len(people)
    print(people.pop(index))

이런 식의 풀이는 어떨까요?

오호.. k번째 사람을 이렇게도 구할 수 있군요 .. ㄷㄷ!! 확실히 코드길이가 줄어드네요
좋은 리뷰 감사합니다 😎