結果

問題 No.2428 Returning Shuffle
ユーザー Manuel1024Manuel1024
提出日時 2023-08-18 22:07:14
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 494 ms / 2,000 ms
コード長 2,178 bytes
コンパイル時間 780 ms
コンパイル使用メモリ 79,408 KB
実行使用メモリ 53,776 KB
最終ジャッジ日時 2023-08-18 22:07:26
合計ジャッジ時間 5,999 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge9
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 314 ms
30,792 KB
testcase_01 AC 494 ms
53,724 KB
testcase_02 AC 462 ms
53,776 KB
testcase_03 AC 25 ms
18,692 KB
testcase_04 AC 26 ms
18,540 KB
testcase_05 AC 27 ms
18,756 KB
testcase_06 AC 26 ms
18,552 KB
testcase_07 AC 27 ms
18,576 KB
testcase_08 AC 28 ms
18,692 KB
testcase_09 AC 29 ms
18,676 KB
testcase_10 AC 28 ms
18,700 KB
testcase_11 AC 28 ms
18,588 KB
testcase_12 AC 27 ms
18,688 KB
testcase_13 AC 27 ms
18,844 KB
testcase_14 AC 27 ms
18,580 KB
testcase_15 AC 26 ms
18,736 KB
testcase_16 AC 28 ms
18,604 KB
testcase_17 AC 25 ms
18,584 KB
testcase_18 AC 25 ms
18,792 KB
testcase_19 AC 335 ms
30,532 KB
testcase_20 AC 338 ms
30,620 KB
testcase_21 AC 28 ms
18,580 KB
testcase_22 AC 26 ms
18,740 KB
testcase_23 AC 27 ms
18,548 KB
testcase_24 AC 342 ms
30,468 KB
testcase_25 AC 333 ms
30,412 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
using ll = long long;
constexpr ll MOD = 998244353;

struct UnionFind{
    vector<int> p;
    UnionFind(int n): p(vector<int>(n, -1)){}

    int root(int x){
        if(p[x] < 0) return x;
        else return p[x] = root(p[x]);
    }

    int size(int x){
        return -p[root(x)];
    }

    void unite(int x, int y){
        x = root(x);
        y = root(y);
        if(x == y) return;
        if(size(x) < size(y)) swap(x, y);
        p[x] += p[y];
        p[y] = x;
    }
};

ll mygcd(ll x, ll y){
    while(y != 0){
        ll r = x%y;
        x = y; y = r;
    }
    return x;
}

int main(){
    int n, m; cin >> n >> m;
    vector<vector<int>> s(m);
    for(int i = 0; i < m; i++){
        int t; cin >> t;
        s[i] = vector<int>(t);
        for(auto &it: s[i]){
            cin >> it;
            it--;
        }
    }

    vector<int> p(n);
    for(int i = 0; i < n; i++){
        p[i] = i;
    }
    for(const auto &v: s){
        int x = p[v.back()];
        for(int i = v.size()-2; i >= 0; i--){
            p[v[i+1]] = p[v[i]];
        }
        p[v[0]] = x;
    }
    // for(int i = 0; i < n; i++) cerr << p[i] << " \n"[i == n-1];

    UnionFind uf(n);
    for(int i = 0; i < n; i++) uf.unite(i, p[i]);

    const int lim = 2000000;
    vector<int> isprime(lim);
    for(int i = 0; i < lim; i++) isprime[i] = i;
    for(int i = 2; i < lim; i++){
        if(isprime[i] != i) continue;
        for(int j = i+i; j < lim; j += i) isprime[j] = min(isprime[j], i);
    }

    vector<int> cnt(lim, 0);
    for(int i = 0; i < n; i++){
        if(uf.root(i) == i){
            int x = uf.size(i);
            // cerr << x << " ";
            while(x > 1){
                int y = isprime[x];
                int z = 0;
                while(x%y == 0){
                    z++;
                    x /= y;
                }
                cnt[y] = max(cnt[y], z);
            }
        }
    }
    ll ans = 1;
    for(int i = 1; i < lim; i++){
        for(int j = 0; j < cnt[i]; j++){
            ans *= i;
            ans %= MOD;
        }
    }
    cout << ans << endl;
    return 0;
}
0