結果

問題 No.3123 Inversion
ユーザー keigo kuwata
提出日時 2025-04-23 08:58:41
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,521 bytes
コンパイル時間 7,264 ms
コンパイル使用メモリ 210,164 KB
実行使用メモリ 65,812 KB
最終ジャッジ日時 2025-04-23 08:59:08
合計ジャッジ時間 22,834 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 1
other AC * 1 WA * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

int main(){
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    int T;
    int mod;
    cin >> T >> mod;
    vector<int> Ns(T);
    int Nmax = 0;
    for(int i = 0; i < T; i++){
        cin >> Ns[i];
        Nmax = max(Nmax, Ns[i]);
    }

    // 前処理
    vector<int> fac(Nmax+1), Inv(Nmax+1);
    fac[0] = 1;
    for(int i = 1; i <= Nmax; i++){
        fac[i] = (int64)fac[i-1] * i % mod;
    }
    Inv[0] = Inv[1] = 1;
    for(int i = 2; i <= Nmax; i++){
        Inv[i] = (Inv[i-1] + (int64)(i-1) * Inv[i-2]) % mod;
    }

    int H = Nmax / 2;
    vector<int> pw2(H+1), Sym(H+1);
    pw2[0] = 1;
    for(int i = 1; i <= H; i++){
        pw2[i] = (pw2[i-1] * 2LL) % mod;
    }
    Sym[0] = 1;
    if(H >= 1) Sym[1] = 2 % mod;
    for(int i = 2; i <= H; i++){
        Sym[i] = (
            2LL * Sym[i-1]
            + 2LL * (i-1) % mod * Sym[i-2]
        ) % mod;
    }

    // 各クエリ処理
    for(int n : Ns){
        if(n == 1){
            cout << 1 << "\n";
            continue;
        }
        int64 F = fac[n];
        int64 D = Inv[n];
        int m = n >> 1;
        int64 C = (int64)fac[m] * pw2[m] % mod;
        int64 A = Sym[m];
        int64 B = ((n%4==0 || n%4==1) ? 2 : 0);

        int64 ans = (
              8LL * F
            - 8LL * D
            - 4LL * C
            + 6LL * A
            - 2LL * B
        ) % mod;
        if(ans < 0) ans += mod;
        cout << ans << "\n";
    }

    return 0;
}
0