結果

問題 No.2045 Two Reflections
ユーザー Carpenters-Cat
提出日時 2022-09-08 01:31:46
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,522 bytes
コンパイル時間 2,146 ms
コンパイル使用メモリ 195,280 KB
最終ジャッジ日時 2025-02-07 03:28:00
ジャッジサーバーID
（参考情報） 		judge5 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 26 WA * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
ll m = 998244353;
ll mpow(ll a, ll n) {
    ll ret = 1;
    while (n) {
        if (n & 1) {
            ret *= a;
            ret %= m;
        }
        n >>= 1;
        a = (a * a) % m;
    }
    return ret;
}
int main () {
    int N, p, q;
    cin >> N >> p >> q;
    if (min(p, q) == N) {
        cout << 2 << endl;
        return 0;
    }
    int A[100010];
    for (int i = 0; i < N; i ++) {
        A[i] = i;
    }
    reverse(A, A + p);
    reverse(A+(N-q), A+N);
    vector<int> did(N, 0);
    vector<int> ans(N + 1, 0);
    vector<int> lpr(N + 1, 0);
    lpr[0] = lpr[1] = 1;
    for (int i = 2; i <= N; i ++) {
        if (!lpr[i]) {
            lpr[i] = i;
            for (int j = 2; j * i <= N; j ++) {
                lpr[i * j] = i;
            }
        }
    }
    for (int i = 0; i < N; i ++) {
        if (did[i]) {
            continue;
        }
        int cnt = 0;
        int fl = i;
        do {
            cnt ++;
            fl = A[fl];
            did[fl] = 1;
        } while (fl != i);
        while (cnt > 1) {
            int c = 0;
            int p = lpr[cnt];
            while (cnt % p == 0) {
                c ++;
                cnt /= p;
            }
            ans[p] = max(ans[p], c);
        }
    }
    ll r = 1;
    for (int i = 2; i <= N; i ++) {
        r *= mpow(i, ans[i]);
        r %= m;
    }
    if (min(p, q) > 1 && max(p, q) < N) {
        r *= 2;
        r %= m;
    }
    cout << r << endl;
}
0