結果

問題 No.1779 Magical Swap
ユーザー kyo1
提出日時 2021-12-09 13:26:06
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 41 ms / 2,000 ms
コード長 987 bytes
コンパイル時間 1,979 ms
コンパイル使用メモリ 201,476 KB
最終ジャッジ日時 2025-01-26 07:15:27
ジャッジサーバーID
(参考情報)
judge4 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 1
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>

using namespace std;

template <typename T>
bool is_prime(T n) {
  if (n == 0 || n == 1) return false;
  if (n == 2) return true;
  if (n % 2 == 0) return false;
  for (T i = 3; i * i < n + 1; i += 2) {
    if (n % i == 0) return false;
  }
  return true;
}

int main() {
  ios::sync_with_stdio(false);
  cin.tie(nullptr);
  int T;
  cin >> T;
  auto solve = [&]() {
    int N;
    cin >> N;
    vector<int> A(N), B(N);
    for (auto &&e : A) {
      cin >> e;
    }
    for (auto &&e : B) {
      cin >> e;
    }
    vector<int> X, Y;
    for (int i = 0; i < N; i++) {
      if (i == 0 || (is_prime(i + 1) && 2 * (i + 1) > N)) {
        if (A[i] != B[i]) {
          cout << "No" << '\n';
          return;
        }
      }
      X.emplace_back(A[i]);
      Y.emplace_back(B[i]);
    }
    sort(X.begin(), X.end());
    sort(Y.begin(), Y.end());
    cout << (X == Y ? "Yes" : "No") << '\n';
  };
  for (int i = 0; i < T; i++) {
    solve();
  }
  return 0;
}
0