結果

問題 No.2428 Returning Shuffle
ユーザー 獅子座じゃない人
提出日時 2023-08-18 22:10:22
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,264 bytes
コンパイル時間 2,498 ms
コンパイル使用メモリ 212,240 KB
最終ジャッジ日時 2025-02-16 10:09:11
ジャッジサーバーID
（参考情報） 		judge4 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 18 WA * 5
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

#include <atcoder/dsu>
#include <atcoder/modint>
using namespace atcoder;
using mint=modint998244353;

int main(void) {
    int n;
    cin >> n;
    int m;
    cin >> m;
    vector<int> t(m);
    vector<vector<int>> s(m);
    vector<int> a(n);
    for(int i=0;i<n;++i){
        a[i]=i;
    }
    for(int i=0;i<m;++i){
        cin >> t[i];
        s[i].resize(t[i]);
        for(int j=0;j<t[i];++j){
            cin >> s[i][j];
            --s[i][j];
        }
        int tmp=a[s[i][t[i]-1]];
        for(int j=t[i]-1;j>0;--j){
            a[s[i][j]]=a[s[i][j-1]];
        }
        a[s[i][0]]=tmp;
    }
    dsu g(n);
    for(int i=0;i<n;++i){
        g.merge(i,a[i]);
    }
    auto groups=g.groups();
    long long ans=1;
    mint answer=1;
    vector<int> sizes;
    map<int,bool> seen;
    for(auto group: groups){
        if(ans*gcd(ans,group.size())>=998244353 && !seen[gcd(ans,group.size())]) {
            seen[gcd(ans,group.size())]=true;
            answer*=gcd(ans,group.size());
        } else if(ans*gcd(ans,group.size())<998244353) {
            seen[gcd(ans,group.size())]=true;
            ans=lcm(ans,group.size());
        }
    }
    answer*=ans;
    cout << answer.val() << endl;
    return 0;
}
0