結果

問題 No.2518 Adjacent Larger
ユーザー cho435
提出日時 2023-10-28 12:57:25
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,247 bytes
コンパイル時間 4,378 ms
コンパイル使用メモリ 253,880 KB
最終ジャッジ日時 2025-02-17 16:38:38
ジャッジサーバーID
（参考情報） 		judge1 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 16 WA * 12
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using ll = long long;
#define rep(i,n) for(int i=0;i<(int)(n);i++)

void solve(){
	int n;
	cin>>n;
	vector<int> a(n);
	rep(i,n) cin>>a.at(i);
	int flg=0;
	rep(i,n-1){
		if(a.at(i)==a.at(i+1)){
			if(a.at(i)!=1) flg=1;
		}
	}
	if(a.at(0)==a.at(n-1)) if(a.at(0)!=1) flg=1;
	int c0=0,c2=0;
	rep(i,n){
		if(a.at(i)==0) c0++;
		if(a.at(i)==2) c2++;
	}
	if(c0*c2==0) flg=1;
	int sm=0;
	rep(i,n) sm+=a.at(i);
	if(sm!=n) flg=1;
	vector<int> ls(n,-1);
	rep(i,n){
		if(a.at(i)==2) ls.at(i)=1;
		if(a.at(i)==0) ls.at(i)=0;
		if(i<n-1){
			if(a.at(i+1)==0) ls.at(i)=1;
			if(a.at(i+1)==2) ls.at(i)=0;
		}
	}
	if(a.at(0)==0) ls.at(n-1)=1;
	if(a.at(0)==2) ls.at(n-1)=0;
	queue<int> q;
	rep(i,n-1){
		if(ls.at(i)!=-1&&ls.at(i+1)==-1) q.push(i+1);
	}
	if(ls.at(n-1)!=-1&&ls.at(0)==-1) q.push(0);
	while(!q.empty()){
		int p=q.front();
		q.pop();
		int bf=(p+n-1)%n;
		if(ls.at(p)!=-1){
			int sm=0;
			sm+=ls.at(p);
			sm+=ls.at(bf)^1;
			if(sm!=a.at(p)){
				flg=1;
				break;
			}
			continue;
		}
		ls.at(p)=a.at(p)-ls.at(bf)^1;
		if(ls.at(p)==2){
			flg=1;
			break;
		}
		q.push((p+1)%n);
	}
	if(flg) cout<<"No\n";
	else cout<<"Yes\n";
}

int main(){
	int t;
	cin>>t;
	rep(i,t) solve();
}
0