結果

問題 No.3109 Swap members
コンテスト
ユーザー takanami
提出日時 2025-04-20 03:23:12
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 491 ms / 2,000 ms
コード長 827 bytes
コンパイル時間 571 ms
コンパイル使用メモリ 12,032 KB
実行使用メモリ 35,584 KB
最終ジャッジ日時 2025-04-20 03:23:28
合計ジャッジ時間 13,749 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 52
権限があれば一括ダウンロードができます

ソースコード

diff # 

def main():
    N, K = map(int, input().split())
    S = [input() for _ in range(N)]
    T = [input() for _ in range(N)]
    
    print(can_rearrange(S, T, N, K))

def can_rearrange(S, T, N, K):
    positions_in_T = {}
    for i in range(N):
        positions_in_T[T[i]] = i + 1
    
    parent = list(range(N+1))
    
    def find(x):
        if parent[x] != x:
            parent[x] = find(parent[x])
        return parent[x]
    
    def union(x, y):
        parent[find(x)] = find(y)
    
    for i in range(1, N-K+1):
        union(i, i+K)
    
    for i in range(N):
        username = S[i]
        initial_position = i + 1
        desired_position = positions_in_T[username]
        if find(initial_position) != find(desired_position):
            return "No"
    
    return "Yes"

if __name__ == "__main__":
    main()
0