[HackerRank] Sequence Equation

[문제 링크]

[입력]

int p[n]: 정수형 배열 , 크기는 n

[입력 제한]

n 의 크기는 1이상 50 이하 이다.
배열 p 의 각 원소는 1이상 50 이하 이다.

[출력]

int[n]: x = 1,2,...,n 일때 p[p[y]] = x 를 만족하는 y를 담은 배열 , 크기는 n 이다.

[코드]

def permutationEquation(p):
    # Write your code here
    result = list()
    for i in range(1,len(p)+1):
        for j in range(1,len(p)+1):
            if(p[p[j-1]-1]==i):
                result.append(j)
                break
    return result