結果

問題 No.2428 Returning Shuffle
ユーザー simansiman
提出日時 2023-08-30 17:40:28
言語 C++17(clang)
(17.0.6 + boost 1.83.0)
結果
AC  
実行時間 294 ms / 2,000 ms
コード長 2,341 bytes
コンパイル時間 6,554 ms
コンパイル使用メモリ 109,024 KB
実行使用メモリ 14,608 KB
最終ジャッジ日時 2023-08-30 17:40:46
合計ジャッジ時間 11,731 ms
ジャッジサーバーID
(参考情報)
judge11 / judge14
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 128 ms
10,176 KB
testcase_01 AC 285 ms
14,460 KB
testcase_02 AC 294 ms
14,608 KB
testcase_03 AC 7 ms
5,256 KB
testcase_04 AC 7 ms
5,260 KB
testcase_05 AC 7 ms
5,232 KB
testcase_06 AC 7 ms
5,216 KB
testcase_07 AC 7 ms
5,236 KB
testcase_08 AC 7 ms
5,252 KB
testcase_09 AC 7 ms
5,252 KB
testcase_10 AC 7 ms
5,268 KB
testcase_11 AC 7 ms
5,228 KB
testcase_12 AC 7 ms
5,332 KB
testcase_13 AC 7 ms
5,228 KB
testcase_14 AC 8 ms
5,252 KB
testcase_15 AC 8 ms
5,300 KB
testcase_16 AC 7 ms
5,260 KB
testcase_17 AC 7 ms
5,236 KB
testcase_18 AC 7 ms
5,196 KB
testcase_19 AC 185 ms
13,560 KB
testcase_20 AC 182 ms
14,088 KB
testcase_21 AC 7 ms
5,216 KB
testcase_22 AC 7 ms
5,244 KB
testcase_23 AC 7 ms
5,224 KB
testcase_24 AC 162 ms
12,060 KB
testcase_25 AC 161 ms
12,100 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

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

using namespace std;
typedef long long ll;

class Prime {
public:
  vector<ll> prime_list;
  const ll MAX_N = 1000000;

  Prime() {
    bool checked[MAX_N + 1];
    memset(checked, false, sizeof(checked));

    for (ll i = 2; i <= MAX_N; ++i) {
      if (!checked[i]) {
        prime_list.push_back(i);

        for (ll j = i * i; j <= MAX_N; j += i) {
          checked[j] = true;
        }
      }
    }
  }

  map<ll, int> prime_division(ll n) {
    map<ll, int> res;

    for (ll i = 0; prime_list[i] <= sqrt(n); ++i) {
      ll p = prime_list[i];

      while (n % p == 0) {
        ++res[p];
        n /= p;
      }
    }

    if (n != 1) {
      res[n] = 1;
    }

    return res;
  }

  bool is_prime(ll n) {
    if (n <= 1) return false;

    for (int i = 0; i < prime_list.size(); ++i) {
      if (n % prime_list[i]) return false;
    }

    return true;
  }
};

int main() {
  int N, M;
  cin >> N >> M;
  int P[N];
  for (int n = 0; n < N; ++n) {
    P[n] = n;
  }

  for (int i = 0; i < M; ++i) {
    int t;
    scanf("%d", &t);
    int S[t];

    for (int j = 0; j < t; ++j) {
      scanf("%d", &S[j]);
    }

    int temp = P[S[0] - 1];

    for (int j = 0; j < t; ++j) {
      int next = S[(j + 1) % t] - 1;

      int old = P[next];
      P[next] = temp;
      temp = old;
    }
  }

  bool visited[N];
  memset(visited, false, sizeof(visited));
  vector<ll> nums;

  for (int v = 0; v < N; ++v) {
    if (visited[v]) continue;

    queue<int> que;
    que.push(v);
    int size = 0;

    while (not que.empty()) {
      int x = que.front();
      que.pop();

      if (visited[x]) continue;
      visited[x] = true;
      ++size;

      que.push(P[x]);
    }

    nums.push_back(size);
  }

  ll MOD = 998244353;
  ll ans = 1;
  Prime prime;
  map<ll, int> values;

  for (ll x : nums) {
    map<ll, int> res = prime.prime_division(x);

    for (auto [n, e] : res) {
      values[n] = max(values[n], e);
    }
  }

  for (auto [n, e] : values) {
    ll v = 1;
    for (int i = 0; i < e; ++i) {
      v *= n;
      v %= MOD;
    }

    ans *= v;
    ans %= MOD;
  }

  cout << ans << endl;

  return 0;
}
0