#853231 (C++14) No.2254 Reverse Only

• 問題: No.2254 Reverse Only
• ユーザー: kotatsugame
• 提出日時: 2023-03-26 15:27:25
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 147 ms / 2,000 ms
• コード長: 959 bytes
• コンパイル時間: 1,668 ms
• コンパイル使用メモリ: 94,324 KB
• 実行使用メモリ: 11,956 KB
• 最終ジャッジ日時: 2024-09-19 09:51:47
• 合計ジャッジ時間: 7,960 ms
• ジャッジサーバーID （参考情報）: judge2 / judge4

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,376 KB
testcase_02	AC	2 ms; 5,376 KB
testcase_03	AC	1 ms; 5,376 KB
testcase_04	AC	2 ms; 5,376 KB
testcase_05	AC	2 ms; 5,376 KB
testcase_06	AC	2 ms; 5,376 KB
testcase_07	AC	2 ms; 5,376 KB
testcase_08	AC	140 ms; 5,376 KB
testcase_09	AC	126 ms; 11,812 KB
testcase_10	AC	124 ms; 5,376 KB
testcase_11	AC	147 ms; 5,376 KB
testcase_12	AC	142 ms; 5,376 KB
testcase_13	AC	110 ms; 5,376 KB
testcase_14	AC	109 ms; 5,376 KB
testcase_15	AC	108 ms; 5,376 KB
testcase_16	AC	127 ms; 11,828 KB
testcase_17	AC	124 ms; 11,832 KB
testcase_18	AC	142 ms; 5,376 KB
testcase_19	AC	132 ms; 5,376 KB
testcase_20	AC	133 ms; 11,956 KB
testcase_21	AC	132 ms; 11,832 KB
testcase_22	AC	73 ms; 11,832 KB
testcase_23	AC	133 ms; 11,832 KB
testcase_24	AC	123 ms; 11,836 KB
testcase_25	AC	135 ms; 11,832 KB
testcase_26	AC	136 ms; 11,816 KB
testcase_27	AC	5 ms; 5,376 KB
testcase_28	AC	33 ms; 5,376 KB
testcase_29	AC	53 ms; 5,376 KB
testcase_30	AC	57 ms; 5,376 KB
testcase_31	AC	27 ms; 5,376 KB
testcase_32	AC	29 ms; 5,376 KB
testcase_33	AC	15 ms; 5,376 KB
testcase_34	AC	68 ms; 5,376 KB
testcase_35	AC	44 ms; 5,376 KB
testcase_36	AC	120 ms; 5,376 KB
testcase_37	AC	126 ms; 5,376 KB
testcase_38	AC	21 ms; 5,376 KB
testcase_39	AC	111 ms; 5,376 KB
testcase_40	AC	5 ms; 5,376 KB
testcase_41	AC	41 ms; 5,376 KB
testcase_42	AC	23 ms; 5,376 KB
testcase_43	AC	73 ms; 5,376 KB
testcase_44	AC	104 ms; 5,376 KB
testcase_45	AC	80 ms; 5,376 KB
testcase_46	AC	4 ms; 5,376 KB

ソースコード

#include<iostream>
#include<vector>
#include<algorithm>
#include<cassert>
#include<atcoder/string>
using namespace std;
int N,K;
int main()
{
	cin>>N>>K;
	vector<int>A(N),B(N);
	for(int i=0;i<N;i++)cin>>A[i];
	for(int i=0;i<N;i++)cin>>B[i];
	if(A==B)
	{
		cout<<"Yes"<<endl;
		return 0;
	}
	if(K==2||K+2<=N)
	{
		sort(A.begin(),A.end());
		sort(B.begin(),B.end());
		if(A==B)cout<<"Yes"<<endl;
		else cout<<"No"<<endl;
		return 0;
	}
	if(K>N)
	{
		cout<<"No"<<endl;
		return 0;
	}
	if(K==N)
	{
		reverse(A.begin(),A.end());
		if(A==B)cout<<"Yes"<<endl;
		else cout<<"No"<<endl;
		return 0;
	}
	assert(K+1==N);
	vector<int>L=B,R=B;
	L.pop_back();
	R.pop_back();
	reverse(R.begin(),R.end());
	for(int t=0;t<2;t++)for(int a:A)
	{
		L.push_back(a);
		R.push_back(a);
	}
	L=atcoder::z_algorithm(L);
	R=atcoder::z_algorithm(R);
	for(int i=0;i<N;i++)if(A[i]==B[N-1])
	{
		if(max(L[N+i],R[N+i])>=N-1)
		{
			cout<<"Yes"<<endl;
			return 0;
		}
	}
	cout<<"No"<<endl;
}