Envío 5160

Código

def convert_to_disjoint_cycles(perm):
    used = {}
    for i in range(len(perm)):    
        if i in used:      
            continue     
        cycle = [i]
        used[i] = (cycle, 0)
        while perm[i] not in used:     
            used[perm[i]] = (cycle, len(cycle))
            cycle.append(perm[i]) 
            i = perm[i]   
    return used

def result(perm, k):
    cycles = convert_to_disjoint_cycles(perm)  
    res = [None for _ in range(len(perm))] 
    for i in range(len(perm)): 
        cyc, idx = cycles[i] 
        j = (idx + (k % len(cyc))) % len(cyc)
        res[cyc[j]] = str(i + 1)
    return res

N, K = map(int, input().split(" "))
arr = [int(i) for i in input().split(" ")]
print(" ".join(result(arr, K)))