题解 | 数列 k 重排

数列 k 重排

https://www.nowcoder.com/practice/e7a39289f41f446e9ec0eaf1198f484d

def solve(testcase):
    n, x = MI()
    A = LII()
    B = LII()

    for i in range(n):
        A[i] -= 1
    
    for bit in range(60):
        if x >> bit & 1:
            B = [B[A[i]] for i in range(n)]
        
        A = [A[A[i]] for i in range(n)]
    
    print(*B)

for testcase in range(1):
    solve(testcase)

全部评论

相关推荐

点赞 评论 收藏
分享
评论
2
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务