結果
問題 | No.2254 Reverse Only |
ユーザー |
👑 ![]() |
提出日時 | 2023-03-24 22:03:55 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 79 ms / 2,000 ms |
コード長 | 3,082 bytes |
コンパイル時間 | 2,102 ms |
コンパイル使用メモリ | 205,440 KB |
最終ジャッジ日時 | 2025-02-11 17:19:05 |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 47 |
ソースコード
#include <bits/stdc++.h> #pragma GCC optimize("Ofast") using namespace std; using std::cout; using std::cin; using std::endl; using ll=long long; using ld=long double; ll ILL=2167167167167167167; const int INF=2100000000; const int mod=998244353; #define rep(i,a,b) for (ll i=a;i<b;i++) #define all(p) p.begin(),p.end() template<class T> using _pq = priority_queue<T, vector<T>, greater<T>>; template<class T> ll LB(vector<T> &v,T a){return lower_bound(v.begin(),v.end(),a)-v.begin();} template<class T> ll UB(vector<T> &v,T a){return upper_bound(v.begin(),v.end(),a)-v.begin();} template<class T> bool chmin(T &a,const T &b){if(a>b){a=b;return 1;}else return 0;} template<class T> bool chmax(T &a,const T &b){if(a<b){a=b;return 1;}else return 0;} template<class T> void So(vector<T> &v) {sort(v.begin(),v.end());} template<class T> void Sore(vector<T> &v) {sort(v.begin(),v.end(),[](T x,T y){return x>y;});} void yneos(bool a){if(a) cout<<"Yes\n"; else cout<<"No\n";} template<class T> void vec_out(vector<T> &p){for(int i=0;i<(int)(p.size());i++){if(i) cout<<" ";cout<<p[i];}cout<<"\n";} template<class T> T vec_min(vector<T> &a){assert(!a.empty());T ans=a[0];for(auto &x:a) chmin(ans,x);return ans;} template<class T> T vec_max(vector<T> &a){assert(!a.empty());T ans=a[0];for(auto &x:a) chmax(ans,x);return ans;} template<class T> T vec_sum(vector<T> &a){assert(!a.empty());T ans=a[0]-a[0];for(auto &x:a) ans+=x;return ans;} namespace po167{ //LCP vec[0,n) and vec[i,n) //for all i //O(vec.size()) template<class T> std::vector<int> Z_algo(std::vector<T> &vec){ int n=vec.size(); int ind=1,j=0,k; std::vector<int> ans(n,0); ans[0]=n; while(ind<n){ while(ind+j<n&&vec[j]==vec[ind+j]) j++; ans[ind]=j; if(j==0){ ind++; continue; } k=1; while(k+ind<n&&ans[k]+k<j){ ans[k+ind]=ans[k]; k++; } j-=k; ind+=k; } return ans; } std::vector<int> Z_algo(std::string &vec){ int n=vec.size(); int ind=1,j=0,k; std::vector<int> ans(n,0); ans[0]=n; while(ind<n){ while(ind+j<n&&vec[j]==vec[ind+j]) j++; ans[ind]=j; if(j==0){ ind++; continue; } k=1; while(k+ind<n&&ans[k]+k<j){ ans[k+ind]=ans[k]; k++; } j-=k; ind+=k; } return ans; } } using po167::Z_algo; void solve(); // oddloop int main() { ios::sync_with_stdio(false); cin.tie(nullptr); int t=1; //cin>>t; rep(i,0,t) solve(); } void solve(){ int N,K; cin>>N>>K; vector<int> A(N),B(N); rep(i,0,N) cin>>A[i]; rep(i,0,N) cin>>B[i]; auto C=A,D=B; So(C),So(D); if(C!=D){ cout<<"No\n"; return; } if(A==B){ cout<<"Yes\n"; return; } if(K>N){ cout<<"No\n"; return; } if(K==N){ reverse(all(A)); yneos(A==B); return; } if(K!=N-1){ cout<<"Yes\n"; return; } vector<int> p; rep(i,0,N) p.push_back(B[i]); p.push_back(-1); rep(i,0,N*2) p.push_back(A[i%N]); auto Z=Z_algo(p); rep(i,0,N){ if(Z[i+1+N]==N){ cout<<"Yes\n"; return; } } cout<<"No\n"; }