結果

問題 No.2975 単調増加部分積
コンテスト
ユーザー zawakasu
提出日時 2024-11-29 22:12:42
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,331 bytes
コンパイル時間 910 ms
コンパイル使用メモリ 95,968 KB
実行使用メモリ 6,824 KB
最終ジャッジ日時 2024-11-29 22:13:00
合計ジャッジ時間 16,565 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 10 WA * 14
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <iomanip>
#include <cassert>
#include <vector>
int main() {
    std::cin.tie(nullptr)->sync_with_stdio(false);
    int N, M, P;
    std::cin >> N >> M >> P;
    std::vector<long long> F(N + 1, 1), invF(N + 1);
    for (int i{1} ; i <= N ; i++) F[i] = (F[i - 1] * i) % P;
    invF[N] = [&](){ // P - 2乗
        long long res{1}, cur{F[N]};
        int exp{P - 2};
        while (exp) {
            if (exp & 1) res = res * cur % P;
            cur = cur * cur % P;
            exp >>= 1;
        }
        return res;
    }();
    for (int i{N} ; i >= 1 ; i--) invF[i - 1] = invF[i] * i % P;
    std::vector<long long> dp(N + 1);
    for (int i{1} ; i <= N ; i++) dp[i] = i;
    long long ans{};
    for (int i{1} ; i <= M ; i++) {
        for (int j{1} ; j <= N ; j++) {
            long long comb{(((F[M] * invF[i]) % P) * invF[M - i]) % P};
            ans += dp[j] * comb;
            ans %= P;
        }
        if (i == M) break;
        std::vector<long long> sum(N + 2);
        for (int i{} ; i < N + 1 ; i++) sum[i + 1] = (sum[i] + dp[i]) % P;
        std::vector<long long> next(N + 1);
        for (int j{1} ; j <= N ; j++) next[j] = (j * sum[j]) % P;
        dp = std::move(next);
    }
    ans *= F[N - M];
    ans %= P;
    ans *= invF[N];
    ans %= P;
    std::cout << ans << '\n';
}
0