結果
問題 | No.1779 Magical Swap |
ユーザー |
![]() |
提出日時 | 2021-12-10 23:00:10 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 218 ms / 2,000 ms |
コード長 | 2,082 bytes |
コンパイル時間 | 3,480 ms |
コンパイル使用メモリ | 185,552 KB |
最終ジャッジ日時 | 2025-01-26 07:31:06 |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 18 |
ソースコード
#include <cstdio> #include <cstring> #include <iostream> #include <string> #include <cmath> #include <bitset> #include <vector> #include <map> #include <set> #include <queue> #include <deque> #include <algorithm> #include <complex> #include <unordered_map> #include <unordered_set> #include <random> #include <cassert> #include <fstream> #include <utility> #include <functional> #include <time.h> #include <stack> #include <array> #include <list> #include <atcoder/all> #define popcount __builtin_popcount using namespace std; using namespace atcoder; typedef long long ll; typedef pair<int, int> P; using mint=modint998244353; struct unionfind{ vector<int> par, sz; unionfind() {} unionfind(int n):par(n), sz(n, 1){ for(int i=0; i<n; i++) par[i]=i; } int find(int x){ if(par[x]==x) return x; return par[x]=find(par[x]); } void unite(int x, int y){ x=find(x); y=find(y); if(x==y) return; if(sz[x]>sz[y]) swap(x, y); par[x]=y; sz[y]+=sz[x]; } bool same(int x, int y){ return find(x)==find(y); } int size(int x){ return sz[find(x)]; } }; int main() { int t;cin>>t; while(t--){ int n; cin>>n; vector<int> a(n+1), b(n+1); for(int i=1; i<=n; i++){ cin>>a[i]; } for(int i=1; i<=n; i++){ cin>>b[i]; } unionfind uf(n+1); for(int k=2; k<=n; k++){ for(int i=k; i<=n-k; i+=k){ uf.unite(i, i+k); } } vector<vector<int>> va(n+1), vb(n+1); for(int i=1; i<=n; i++){ int r=uf.find(i); va[r].push_back(a[i]); vb[r].push_back(b[i]); } bool ok=1; for(int i=1; i<=n; i++){ if(va[i].empty()) continue; sort(va[i].begin(), va[i].end()); sort(vb[i].begin(), vb[i].end()); if(va[i]!=vb[i]){ ok=0;break; } } cout<<(ok?"Yes":"No")<<endl; } return 0; }