[파이썬] 문제 : N일 후 작업자 찾기

 

lst = ['A','B','C','D']

N = 9            # N일 지정

if N <= 4:       # N이 4보다 작거나 같으면
    d = N - 1    # N - 1 로 함. 리스트의 인덱스는 0부터 시작하기때문임.
else:
    d = N%4 - 1  # N이 4보다 크면 N을 4로 나눈 나머지에 -1 함.

print(lst[d])

 

검증

# 검증
lst = ['A','B','C','D']

N = 9            # N일 지정

for i in range(1, N+1):
    n = i
    if n <= 4:       # N이 4보다 작거나 같으면
        d = n - 1    # N - 1 로 함. 리스트의 인덱스는 0부터 시작하기때문임.
    else:
        d = n%4 - 1  # N이 4보다 크면 N을 4로 나눈 나머지에 -1 함.

    print(i, lst[d])