結果

問題 No.2254 Reverse Only
コンテスト
ユーザー lddlinan
提出日時 2023-03-25 14:45:28
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 176 ms / 2,000 ms
コード長 2,716 bytes
コンパイル時間 1,125 ms
コンパイル使用メモリ 94,552 KB
最終ジャッジ日時 2025-02-11 18:03:58
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 47
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:34:10: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   34 |     scanf("%d %d", &n, &k);
      |     ~~~~~^~~~~~~~~~~~~~~~~
main.cpp:36:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   36 |         scanf("%d", &as[i]);
      |         ~~~~~^~~~~~~~~~~~~~
main.cpp:39:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   39 |         scanf("%d", &bs[i]);
      |         ~~~~~^~~~~~~~~~~~~~

ソースコード

diff # 

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include <map>
#include <vector>
#include <queue>
#include <deque>
#include <set>
#include <stack>
#include <algorithm>
#include <array>
#include <unordered_set>
#include <unordered_map>
#include <string>
using namespace std;

bool rcmp(int a, int b) { return a>b; }
typedef long long LL;
class mypcmp {
public:
    bool operator()(const int& a, const int& b) {
        return a<b;
    }
};



int as[200004];
int bs[600004];
int zf[400004];
int main() {
    int n, i, k, v, t, s, e;
    int c, ze, zc, pe, pi, pc;
    scanf("%d %d", &n, &k);
    for (i=0; i<n; i++) {
        scanf("%d", &as[i]);
    }
    for (i=0; i<n; i++) {
        scanf("%d", &bs[i]);
    }
    map<int, int> cs;
    for (i=0; i<n; i++) cs[as[i]]++;
    for (i=0; i<n; i++) {
        v=bs[i];
        if (cs[v]==0) break;
        cs[v]--;
    }
    if(i<n) printf("No\n");
    else {
        for (i=0; i<n; i++) if (as[i]!=bs[i]) break;
        if (i>=n) { printf("Yes\n"); return 0; }
        if (k>n) { printf("No\n"); return 0; }
        for (i=0; i<n; i++) if (as[i]!=bs[n-1-i]) break;
        if (i>=n) { printf("Yes\n"); return 0; }
        if (k==n) { printf("No\n"); return 0; }
        if (k==n-1) {
            for (i=0; i<n; i++) bs[i+n]=bs[i+n+n]=as[i];
            e=-1; c=0;
            zf[0]=0;
            for (i=1; i<n+n; i++) {
                if (e<=i) { ze=i; zc=0; }
                else {
                    // [e-c, e)
                    s=e-c; pi=i-s; pe=zf[pi]; pc=pe-pi;
                    ze=i+pc; if (ze>e) ze=e;
                    zc=ze-i;
                }
                while(ze<n+n+n&&bs[ze]==bs[zc]) { ze++; zc++; }
                if (i>=n&&zc>=n) {
                    break;
                }
                if (ze>e) { e=ze; c=zc; }
                zf[i]=ze;
            }
            if (i<n+n) { printf("Yes\n"); return 0; }
            s=0; e=n-1; while(s<e) {
                t=bs[s]; bs[s]=bs[e]; bs[e]=t;
                s++; e--;
            }
            e=-1; c=0;
            for (i=1; i<n+n; i++) {
                if (e<=i) { ze=i; zc=0; }
                else {
                    // [e-c, e)
                    s=e-c; pi=i-s; pe=zf[pi]; pc=pe-pi;
                    ze=i+pc; if (ze>e) ze=e;
                    zc=ze-i;
                }
                while(ze<n+n+n&&bs[ze]==bs[zc]) { ze++; zc++; }
                if (i>=n&&zc>=n) {
                    break;
                }
                if (ze>e) { e=ze; c=zc; }
                zf[i]=ze;
            }
            if (i<n+n) { printf("Yes\n"); return 0; }
            printf("No\n");
            return 0;
        }
        printf("Yes\n");
    }
    return 0;
}
0