結果

問題 No.2254 Reverse Only
コンテスト
ユーザー sotanishy
提出日時 2023-03-24 21:52:40
言語 PyPy3
(8.0.0 + ACL)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
WA  
(最新)
AC  
(最初)
実行時間 -
コード長 924 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 338 ms
コンパイル使用メモリ 64,628 KB
実行使用メモリ 100,980 KB
最終ジャッジ日時 2026-10-24 22:21:04
合計ジャッジ時間 7,715 ms
ジャッジサーバーID
(参考情報)
judge4_0 / judge3_0
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 48 WA * 1
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code

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