結果

問題 No.3109 Swap members
ユーザー koufu193
提出日時 2025-04-19 09:26:47
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 961 bytes
コンパイル時間 1,200 ms
コンパイル使用メモリ 82,156 KB
実行使用メモリ 108,228 KB
最終ジャッジ日時 2025-04-19 09:27:01
合計ジャッジ時間 11,789 ms
ジャッジサーバーID
(参考情報)
judge2 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 33 WA * 19
権限があれば一括ダウンロードができます

ソースコード

diff #

def can_sort(N, K, S, T):
    # グループを管理するためにvisitedリストを使用
    visited = [False] * N
    groups = []
    
    # グループ分け
    for i in range(N):
        if not visited[i]:
            group = []
            j = i
            while not visited[j]:
                visited[j] = True
                group.append(j)
                j = (j + K) % N  # 次に移動する位置
            groups.append(group)
    
    # 各グループでSとTが一致するかを確認
    for group in groups:
        # S[group]とT[group]をソートして一致するか確認
        s_group = sorted(S[i] for i in group)
        t_group = sorted(T[i] for i in group)
        if s_group != t_group:
            return "No"
    
    return "Yes"


# 入力の読み込み
N, K = map(int, input().split())
S = [input().strip() for _ in range(N)]
T = [input().strip() for _ in range(N)]

# 結果を出力
print(can_sort(N, K, S, T))
0