結果

問題 No.3109 Swap members
ユーザー Ka1chiX_m
提出日時 2025-04-19 05:27:24
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 228 ms / 2,000 ms
コード長 597 bytes
コンパイル時間 886 ms
コンパイル使用メモリ 12,416 KB
実行使用メモリ 66,624 KB
最終ジャッジ日時 2025-04-19 05:27:33
合計ジャッジ時間 7,658 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 52
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import defaultdict

def can_sort_with_k_swaps(N, K, S, T):
    s_groups = defaultdict(list)
    t_groups = defaultdict(list)

    for i in range(N):
        s_groups[i % K].append(S[i])
        t_groups[i % K].append(T[i])

    for r in range(K):
        if sorted(s_groups[r]) != sorted(t_groups[r]):
            return False
    return True

def main():
    import sys
    input = sys.stdin.read
    data = input().split()

    N = int(data[0])
    K = int(data[1])
    S = data[2:N+2]
    T = data[N+2:]

    print("Yes" if can_sort_with_k_swaps(N, K, S, T) else "No")

main()
0