結果

問題 No.2428 Returning Shuffle
コンテスト
ユーザー nono00
提出日時 2023-08-18 22:40:33
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 249 ms / 2,000 ms
コード長 2,155 bytes
コンパイル時間 4,181 ms
コンパイル使用メモリ 258,856 KB
実行使用メモリ 10,880 KB
最終ジャッジ日時 2024-11-28 08:45:01
合計ジャッジ時間 6,573 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 23
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

namespace nono {

constexpr long long mod = 998244353;

std::vector<std::pair<int, int>> factorize(int n) {
    std::vector<std::pair<int, int>> result;
    for (long long i = 2; i * i <= n; i++) {
        int count = 0;
        while (n % i == 0) {
            n /= i;
            count++;
        }
        if (count > 0) {
            result.emplace_back(i, count);
        }
    }
    if (n != 1) {
        result.emplace_back(n, 1);
    }
    return result;
}

long long pow(long long base, long long exp, long long mod) {
    long long result = 1;
    while (exp > 0) {
        if (exp & 1) {
            result *= base;
            result %= mod;
        }
        base *= base;
        base %= mod;
        exp >>= 1;
    }
    return result;
}

void solve() {
    int n, m;
    std::cin >> n >> m;
    std::vector<int> a(n);
    std::iota(a.begin(), a.end(), 0);
    for (int i = 0; i < m; i++) {
        int t;
        std::cin >> t;
        std::vector<int> s(t);
        for (int j = 0; j < t; j++) {
            std::cin >> s[j];
            s[j]--;
        }
        int temp = a[s.back()];
        for (int j = t - 1; j > 0; j--) {
            a[s[j]] = a[s[j - 1]];
        }
        a[s.front()] = temp;
    }

    //  for (int i = 0; i < n; i++) {
    //      std::cout << a[i] << ' ';
    //  }
    //  std::cout << std::endl;

    std::vector<bool> used(n);
    std::map<int, int> mp;
    for (int i = 0; i < n; i++) {
        if (not used[i]) {
            int s = i;
            int len = 0;
            while (not used[s]) {
                used[s] = true;
                s = a[s];
                len++;
            }
            for (const auto& [p, v]: factorize(len)) {
                mp[p] = std::max(mp[p], v);
            }
        }
    }

    long long ans = 1;
    for (const auto& [p, v]: mp) {
        ans *= pow(p, v, mod);
        ans %= mod;
    }
    std::cout << ans << std::endl;
}

}  //  namespace nono

int main() {
    std::cin.tie(0)->sync_with_stdio(0);
    std::cout << std::fixed << std::setprecision(15);

    int t = 1;
    //  std::cin >> t;
    while (t--) nono::solve();
}
0