結果

問題 No.2346 Replace!!
コンテスト
ユーザー ytqm3
提出日時 2023-06-04 01:12:10
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
(最新)
AC  
(最初)
実行時間 -
コード長 1,792 bytes
コンパイル時間 4,298 ms
コンパイル使用メモリ 259,768 KB
最終ジャッジ日時 2025-02-13 22:30:39
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 1
other AC * 62 WA * 11
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

bool is_subseq(vector<int> a,vector<int> b){
	//b は a の部分列か？
	int idx=0;
	for(int i=0;i<a.size();++i){
		if(a[i]==b[idx]){ idx++; }
		if(idx==b.size()){ return 1; }
	}
	return 0;
}

void solve(){
  int N;
  cin>>N;
  vector<int> A(N),B(N);
  for(auto& v:A){ cin>>v; v--; }
  for(auto& v:B){ cin>>v; v--; }
  //V : サイクルとして採用したか
  vector<int> V(N);
  //S : サイクルの集合
  vector<vector<int>> S;
  for(int i=0;i<N;++i){
    int nw=i,f=0;
    for(int j=0;j<N;++j){
      if(nw==B[i]){ f=1; }
      nw=A[nw];
      if(nw==i && V[i]==0){
        vector<int> T;
        auto tmp=i;
        for(int k=0;k<=j;++k){
          T.emplace_back(tmp);
          V[tmp]=1;
          tmp=A[tmp];
        }
  			assert(tmp==i);
        S.emplace_back(T);
      }
    }
    if(!f){
    	cout<<"No"<<endl;
      return;
    }
  }
  for(auto v:S){
  	int n=v.size();
  	//v0 -> v1 -> ... -> v{n-1} -> v0
  	//vi - B[vi] が連結か判定
  	atcoder::dsu uf(N);
  	for(int i=0;i<n;++i){
  		uf.merge(v[i],B[v[i]]);
  	}
  	if(uf.groups().size()!=N-n+1){
  		cout<<"No"<<endl;
  		return;
  	}
  	//vi -> B[vi] を張ったグラフにおけるサイクルを検出
  	vector<int> w;
  	for(int i=0;i<n;++i){
  		int nw=v[i];
  		for(int j=0;j<n;++j){
  			nw=B[nw];
  			if(nw==v[i]){
  				int tmp=v[i];
  				for(int k=0;k<=j;++k){
  					w.emplace_back(tmp);
  					tmp=B[tmp];
  				}
  				assert(tmp==v[i]);
  				goto check;
  			}
  		}
  	}
  	check:
  	//w0 は w のうち v で最初にでてくるもの
  	if(!is_subseq(v,w)){
  		cout<<"No"<<endl;
  		return;
  	}
  }
  cout<<"Yes"<<endl;
}

int main(){
  int T;
  cin>>T;
  while(T--){
    solve();
  }
}
0