2018년 2월 6일 화요일

[python][hackerrank] Sequence Equation

처음에는 리스트로 그냥 뿌려서 하면 되는 건가 했는데
해(OUTPUT)를 가지고 반대로 x값(INPUT)을 구하는 것이었다.
값이 없는 경우가 있다면 그것들을 어떻게 해야 하는지 설명이 필요한데 그에 대한 설명이 없어서
그런경우는 없다는 가정하에 문제를 풀었다.

그래서 미리 p(p(y))를 풀어서 key값에 넣고 그 안에 x(INPUT)값을 value로 넣었다.

n =  int(input())
n_list = list(map(int, input().strip().split(' ')))
p = { n_list[n_list[x-1]-1]:x for x in n_list}

for i in range(1, n+1):
    print(p[i])

댓글 없음 :

댓글 쓰기