結果
問題 |
No.3109 Swap members
|
ユーザー |
|
提出日時 | 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 |
ソースコード
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()