#918752 (C++14) No.2499 Sum of Products of Sums

• 問題: No.2499 Sum of Products of Sums
• ユーザー: kotatsugame
• 提出日時: 2023-10-06 23:11:31
• 言語: C++14 (gcc 13.3.0 + boost 1.87.0)
• 結果: AC
• 実行時間: 115 ms / 2,000 ms
• コード長: 1,628 bytes
• コンパイル時間: 2,230 ms
• コンパイル使用メモリ: 114,692 KB
• 実行使用メモリ: 6,944 KB
• 最終ジャッジ日時: 2024-07-26 16:56:22
• 合計ジャッジ時間: 3,619 ms
• ジャッジサーバーID （参考情報）: judge2 / judge5

ファイルパターン	結果
sample	AC * 3
other	AC * 17

ソースコード

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
#include<iostream>
#include<cassert>
#include<atcoder/modint>
#include<atcoder/convolution>
using namespace std;
#include<vector>
template<typename T>
struct combination{
    vector<T>fac,ifac;
    combination(size_t N=0):fac(1,1),ifac(1,1)
    {
        make_table(N);
    }
    void make_table(size_t N)
    {
        if(fac.size()>N)return;
        size_t now=fac.size();
        N=max(N,now*2);
        fac.resize(N+1);
        ifac.resize(N+1);
        for(size_t i=now;i<=N;i++)fac[i]=fac[i-1]*i;
        ifac[N]=1/fac[N];
        for(size_t i=N;i-->now;)ifac[i]=ifac[i+1]*(i+1);
    }
    T factorial(size_t n)
    {
        make_table(n);
        return fac[n];
    }
    T invfac(size_t n)
    {
        make_table(n);
        return ifac[n];
    }
    T P(size_t n,size_t k)
    {
        if(n<k)return 0;
        make_table(n);
        return fac[n]*ifac[n-k];
    }
    T C(size_t n,size_t k)
    {
        if(n<k)return 0;
        make_table(n);
        return fac[n]*ifac[n-k]*ifac[k];
    }
    T H(size_t n,size_t k)
    {
        if(n==0)return k==0?1:0;
        return C(n-1+k,k);
    }
};
using mint=atcoder::modint998244353;
combination<mint>C;
mint comb(int A,int B)
{
    assert(B>=0);
    if(A<B)return mint(0);
    mint x=1,y=1;
    for(int i=1;i<=B;i++)
    {
        x*=A+1-i;
        y*=mint::raw(i);
    }
    return x/y;
}
int H,W;
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    cin>>H>>W;
    vector<mint>dp(1,1);
    for(int i=0;i<H;i++)
    {
        int A,B;cin>>A>>B;
        vector<mint>now(W+1);
        mint CB=comb(B+W,W);
        mint CA=comb(A-1+W,W);
        for(int j=0;j<=W;j++)
        {
            //CB=C(B+W,W+j)
            now[j]=CB-CA;
            mint iv=C.factorial(W+j)*C.invfac(W+j+1);
            CB=CB*(B-j)*iv;
            CA=CA*(A-1-j)*iv;
            now[j]*=C.invfac(j);
        }
        dp=atcoder::convolution(dp,now);
        dp.resize(W+1);
    }
    cout<<(dp[W]*C.factorial(W)).val()<<endl;
}
 
 
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX