結果

問題 No.2254 Reverse Only
コンテスト
ユーザー rin204
提出日時 2024-04-29 14:42:04
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 944 bytes
コンパイル時間 392 ms
コンパイル使用メモリ 82,408 KB
実行使用メモリ 167,256 KB
最終ジャッジ日時 2024-11-18 19:01:21
合計ジャッジ時間 8,055 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 42 WA * 5
権限があれば一括ダウンロードができます

ソースコード

diff # 

def Z_algorithm(S):
    le = len(S)
    Z = [0] * le
    Z[0] = le
    i = 1
    j = 0
    while i < le:
        while i + j < le and S[j] == S[i + j]:
            j += 1
        Z[i] = j
        if j == 0:
            i += 1
            continue

        k = 1
        while i + k < le and k + Z[k] < j:
            Z[i + k] = Z[k]
            k += 1
        i += k
        j -= k
    return Z


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

if A == B:
    print("Yes")
elif n < k:
    print("No")
elif n == k:
    if A == B[::-1]:
        print("Yes")
    else:
        print("No")
elif n == k + 1:
    C = B + A + A
    D = B + A[::-1] + A[::-1]
    Z1 = Z_algorithm(C)
    Z2 = Z_algorithm(D)
    ma = max(*Z1[:n:], *Z2[n:])
    if ma >= n:
        print("Yes")
    else:
        print("No")
else:
    if sorted(A) == sorted(B):
        print("Yes")
    else:
        print("No")
0