結果

問題 No.2254 Reverse Only
ユーザー kuroni
提出日時 2023-04-06 06:22:45
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 72 ms / 2,000 ms
コード長 1,849 bytes
コンパイル時間 1,782 ms
コンパイル使用メモリ 173,332 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-10-02 07:51:20
合計ジャッジ時間 5,898 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 47
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
    int n, k; cin >> n >> k;
    vector<int> a(n), kmp(n + 1);
    for (int i = 0; i < n; i++) {
        cin >> a[i];
    }
    for (int i = 2; i <= n; i++) {
        int &u = kmp[i];
        for (u = i - 1; u > 0 && a[u] != a[i - 1]; u = kmp[u]);
        u += (u == 0 && a[u] == a[i - 1]);
    }
    vector<int> b(n);
    for (int i = 0; i < n; i++) {
        cin >> b[i];
    }
    bool ok = false;
    if (k > n) {
        ok |= (a == b);
    } else if (k == n) {
        ok = (a == b);
        reverse(b.begin(), b.end());
        ok |= (a == b);
    } else if (k == n - 1) {
        {
            int x = 0;
            for (int i = 0; i < n; i++) {
                while (x == n || (x > 0 && a[x] != b[i])) {
                    x = kmp[x];
                }
                x += (a[x] == b[i]);
            }
            for (int i = 0; i < n; i++) {
                while (x == n || (x > 0 && a[x] != b[i])) {
                    x = kmp[x];
                }
                x += (a[x] == b[i]);
                ok |= (x == n);
            }
        }
        reverse(a.begin(), a.end());
        {
            int x = 0;
            for (int i = 0; i < n; i++) {
                while (x == n || (x > 0 && a[x] != b[i])) {
                    x = kmp[x];
                }
                x += (a[x] == b[i]);
            }
            for (int i = 0; i < n; i++) {
                while (x == n || (x > 0 && a[x] != b[i])) {
                    x = kmp[x];
                }
                x += (a[x] == b[i]);
                ok |= (x == n);
            }
        }
    } else {
        sort(a.begin(), a.end());
        sort(b.begin(), b.end());
        ok |= (a == b);
    }
    cout << (ok ? "Yes\n" : "No\n");
}
0