글 읽기 - 어디에서 틀린지 모르겠습니다

hello70825 3년 전 0

틀린 부분이 없는 것 같아서 인터넷에 코드를 검색해봤는데, 18줄 ~ 21줄 부분만 살짝 다르고 로직은 똑같았습니다

학생 번호와 유튜브 채널 번호를 1이 아닌 0부터 시작하는 것으로 만들었고, 답을 구할 때는 1을 더해줘서 ans에 저장하게 만들었습니다.

어디에서 틀린걸까요 ㅠㅠ

input = __import__('sys').stdin.readline
max_p = 30
n,k,m= map(int,input().split())
st = [*map(int,input().split())] # student
yt = [*map(int,input().split())] # youtube
pt = [[-1]*max_p for _ in range(k)] # parent
ans = [] # 출력

for i in range(k):
    pt[i][0] = yt[i]-1
for j in range(max_p-1):
    for i in range(n):
        pt[i][j+1] = pt[pt[i][j]][j]

for i in range(n):
    x = st[i]-1
    diff = m-1
    for j in range(max_p-1,-1,-1):
        if diff >= 2**j:
            diff -= 2**j
            x = pt[x][j]
    ans.append(x+1)
print(*ans)

hello70825 3년 전 0

글 쓰고 다시 보니까 바로 찾았네요..

k를 썼어야 했는데 n을 적었습니다

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

어디에서 틀린지 모르겠습니다

hello70825 3년 전 0

hello70825 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말