結果

問題 No.1191 数え上げを愛したい（数列編）
コンテスト
ユーザー Chihaya_chan
提出日時 2020-09-03 01:04:48
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 116 ms / 2,000 ms
コード長 1,055 bytes
コンパイル時間 238 ms
コンパイル使用メモリ 82,432 KB
実行使用メモリ 80,512 KB
最終ジャッジ日時 2024-11-22 04:50:51
合計ジャッジ時間 3,367 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

class Combi():
    def __init__(self, N, mod):
        self.power = [1 for _ in range(N+1)]
        self.rev = [1 for _ in range(N+1)]
        self.mod = mod
        for i in range(2, N+1):
            self.power[i] = (self.power[i-1]*i) % self.mod
        self.rev[N] = pow(self.power[N], self.mod-2, self.mod)
        for j in range(N, 0, -1):
            self.rev[j-1] = (self.rev[j]*j) % self.mod

    def com(self, K, R):
        if K < R:
            return 0
        else:
            return ((self.power[K])*(self.rev[K-R])*(self.rev[R])) % self.mod

    def pom(self, K, R):
        if K < R:
            return 0
        else:
            return (self.power[K])*(self.rev[K-R]) % self.mod


def main():
    mod = 998244353
    N, M, A, B = map(int, input().split())
    Com = Combi(3*10**5, mod)
    ans = 0
    for k in range(A, B+1):
        if k >= A*(N-1):
            ans += Com.com(k-(A-1)*(N-1)-1, N-2)*(M-k)
    for i in range(1, N+1):
        ans *= i
        ans %= mod
    print(ans)
    return


if __name__ == "__main__":
    main()
0