結果

問題 No.1779 Magical Swap
ユーザー siman
提出日時 2021-12-08 07:08:09
言語 C++17(clang)
(17.0.6 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 1,229 bytes
コンパイル時間 3,936 ms
コンパイル使用メモリ 141,992 KB
実行使用メモリ 13,756 KB
最終ジャッジ日時 2024-07-08 12:18:15
合計ジャッジ時間 5,070 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other TLE * 1 -- * 17
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <cassert>
#include <cmath>
#include <algorithm>
#include <iostream>
#include <iomanip>
#include <climits>
#include <map>
#include <queue>
#include <set>
#include <cstring>
#include <vector>

using namespace std;
typedef long long ll;

int main() {
  int T;
  cin >> T;

  for (int t = 0; t < T; ++t) {
    int N;
    cin >> N;
    vector<int> A(N);
    vector<int> B(N);

    for (int i = 0; i < N; ++i) {
      cin >> A[i];
    }
    for (int i = 0; i < N; ++i) {
      cin >> B[i];
    }

    vector<int> A_copy = A;
    vector<int> B_copy = B;
    sort(A_copy.begin(), A_copy.end());
    sort(B_copy.begin(), B_copy.end());

    if (A[0] != B[0] || A_copy != B_copy) {
      cout << "No" << endl;
    } else {
      int i = 3;
      bool ok = true;
      vector<bool> checked(N + 1, false);

      while (i <= N && ok) {
        if (checked[i]) continue;

        if (2 * i <= N) {
          for (int j = i; j <= N; j += i) {
            checked[j] = true;
          }
        }

        if (not checked[i] && A[i - 1] != B[i - 1]) {
          ok = false;
        }

        i += 2;
      }

      if (ok) {
        cout << "Yes" << endl;
      } else {
        cout << "No" << endl;
      }
    }
  }

  return 0;
}
0