結果

問題 No.2254 Reverse Only
ユーザー jiangly
提出日時 2023-03-24 21:54:17
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 233 ms / 2,000 ms
コード長 1,523 bytes
コンパイル時間 1,810 ms
コンパイル使用メモリ 198,984 KB
最終ジャッジ日時 2025-02-11 17:15:33
ジャッジサーバーID
(参考情報)
judge3 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 47
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>

using i64 = long long;

int main() {
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);
    
    int N, k;
    std::cin >> N >> k;
    
    std::vector<int> A(N), B(N);
    for (int i = 0; i < N; i++) {
        std::cin >> A[i];
    }
    for (int i = 0; i < N; i++) {
        std::cin >> B[i];
    }
    
    if (A == B) {
        std::cout << "Yes\n";
        return 0;
    }
    
    if (k > N) {
        std::cout << "No\n";
        return 0;
    }
    
    std::reverse(A.begin(), A.end());
    
    if (A == B) {
        std::cout << "Yes\n";
        return 0;
    }
    
    if (k == N) {
        std::cout << "No\n";
        return 0;
    }
    
    if (std::multiset(A.begin(), A.end()) != std::multiset(B.begin(), B.end())) {
        std::cout << "No\n";
        return 0;
    }
    
    if (k + 2 <= N) {
        std::cout << "Yes\n";
        return 0;
    }
    
    std::vector<int> f(N + 1);
    for (int i = 1, j = 0; i < N; i++) {
        while (j && B[i] != B[j]) {
            j = f[j];
        }
        j += (B[i] == B[j]);
        f[i + 1] = j;
    }
    for (int t = 0; t < 2; t++) {
        for (int i = 0, j = 0; i < 2 * N; i++) {
            while (j && A[i % N] != B[j]) {
                j = f[j];
            }
            j += (A[i % N] == B[j]);
            if (j == N) {
                std::cout << "Yes\n";
                return 0;
            }
        }
        std::reverse(A.begin(), A.end());
    }
    std::cout << "No\n";
    
    return 0;
}
0