結果

問題 No.1693 Invasion
ユーザー tcltk
提出日時 2021-10-16 04:55:12
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 232 ms / 2,000 ms
コード長 1,094 bytes
コンパイル時間 173 ms
コンパイル使用メモリ 82,348 KB
実行使用メモリ 108,340 KB
最終ジャッジ日時 2024-09-17 19:17:33
合計ジャッジ時間 5,123 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

#!/usr/bin/env python3
# from typing import *

import sys
import io
import math
import collections
import decimal
import itertools
import bisect
import heapq


def input():
    return sys.stdin.readline()[:-1]


# sys.setrecursionlimit(1000000)

# _INPUT = """3 141
# 5 9 2

# """
# sys.stdin = io.StringIO(_INPUT)

INF = 10**10

MOD = 998244353


def comb(n, r):
    r = min(r, n - r)
    return (fact[n] * factinv[r] * factinv[n-r]) % MOD

N, M = map(int, input().split())
A = list(map(int, input().split()))


fact = [1, 1]  # fact[n] = (n! mod p)
factinv = [1, 1]  # factinv[n] = ((n!)^(-1) mod p)
inv = [0, 1]  # factinv 計算用

for i in range(2, M + 1):
    fact.append((fact[-1] * i) % MOD)
    inv.append((-inv[MOD % i] * (MOD // i)) % MOD)
    factinv.append((factinv[-1] * inv[-1]) % MOD)


dp = [INF] * (M+1)
dp[0] = 0

for m in range(M+1):
    for i in range(N):
        if m+A[i] <= M:
            dp[m+A[i]] = min(dp[m+A[i]], dp[m] + 1)

ans = 0
for m in range(M+1):
    if dp[m] < INF:
        # print(m, dp[N][m])
        ans = (ans + comb(M-dp[m], m-dp[m])) % MOD
print(ans)
0