結果

問題 No.2254 Reverse Only
ユーザー sotanishy
提出日時 2023-03-24 21:52:40
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 197 ms / 2,000 ms
コード長 924 bytes
コンパイル時間 194 ms
コンパイル使用メモリ 82,248 KB
実行使用メモリ 135,072 KB
最終ジャッジ日時 2024-09-18 17:02:19
合計ジャッジ時間 7,434 ms
ジャッジサーバーID
(参考情報)
judge3 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 47
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys
input = sys.stdin.readline

def z_array(s):
    n = len(s)
    z = [0] * n
    z[0] = n
    l = r = 0
    for i in range(1, n):
        k = i - l
        if i <= r and z[k] < r - i + 1:
            z[i] = z[k]
        else:
            l = i
            if i > r:
                r = i
            while r < n and s[r - l] == s[r]:
                r += 1
            r -= 1
            z[i] = r - l + 1
    return z


N, k = map(int, input().split())
A = list(map(int, input().split()))
B = list(map(int, input().split()))

if sorted(A) != sorted(B):
    print("No")
elif A == B:
    print("Yes")
elif k > N:
    print("No")
elif k <= N-2:
    # rotation + swap
    print("Yes")
elif k == N-1:
    # rotation
    z = z_array(B + [0] + A + A)
    for i in range(N):
        if z[N+1+i] >= N:
            print("Yes")
            exit()
    print("No")
else:
    # k == N
    print("Yes" if A[::-1] == B else "No")
0