結果

問題 No.2141 Enumeratest
ユーザー 👑 KazunKazun
提出日時 2022-12-02 21:25:35
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 102 ms / 2,000 ms
コード長 4,629 bytes
コンパイル時間 276 ms
コンパイル使用メモリ 82,436 KB
実行使用メモリ 106,752 KB
最終ジャッジ日時 2024-10-09 22:29:14
合計ジャッジ時間 3,887 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 39 ms
52,608 KB
testcase_01 AC 38 ms
52,480 KB
testcase_02 AC 54 ms
65,280 KB
testcase_03 AC 61 ms
74,368 KB
testcase_04 AC 102 ms
106,752 KB
testcase_05 AC 73 ms
90,368 KB
testcase_06 AC 42 ms
52,480 KB
testcase_07 AC 45 ms
60,160 KB
testcase_08 AC 53 ms
64,512 KB
testcase_09 AC 49 ms
62,592 KB
testcase_10 AC 54 ms
67,456 KB
testcase_11 AC 63 ms
72,960 KB
testcase_12 AC 45 ms
59,392 KB
testcase_13 AC 49 ms
62,208 KB
testcase_14 AC 61 ms
72,832 KB
testcase_15 AC 54 ms
66,688 KB
testcase_16 AC 59 ms
69,120 KB
testcase_17 AC 82 ms
89,856 KB
testcase_18 AC 83 ms
90,368 KB
testcase_19 AC 93 ms
99,584 KB
testcase_20 AC 84 ms
89,344 KB
testcase_21 AC 85 ms
92,032 KB
testcase_22 AC 76 ms
82,560 KB
testcase_23 AC 67 ms
73,088 KB
testcase_24 AC 75 ms
81,664 KB
testcase_25 AC 56 ms
65,920 KB
testcase_26 AC 80 ms
89,600 KB
testcase_27 AC 83 ms
89,728 KB
testcase_28 AC 88 ms
95,488 KB
testcase_29 AC 71 ms
79,488 KB
testcase_30 AC 92 ms
99,584 KB
testcase_31 AC 92 ms
87,424 KB
testcase_32 AC 99 ms
92,544 KB
testcase_33 AC 88 ms
84,864 KB
testcase_34 AC 98 ms
89,984 KB
testcase_35 AC 56 ms
62,976 KB
testcase_36 AC 100 ms
94,464 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

"""
Mod はグローバル変数からの指定とする.
"""

"""
積
"""
def product_modulo(*X):
    y=1
    for x in X:
        y=(x*y)%Mod
    return y

"""
階乗
"""
def Factor(N):
    """ 0!, 1!, ..., N! (mod Mod) を出力する.

    N: int
    """
    f=[1]*(N+1)
    for k in range(1,N+1):
        f[k]=(k*f[k-1])%Mod
    return f

def Factor_with_inverse(N):
    """ 0!, 1!, ..., N!, (0!)^-1, (1!)^-1, ..., (N!)^-1 を出力する.

    N: int
    """

    f=Factor(N)
    g=[0]*(N+1)

    N=min(N,Mod-1)
    g[N]=pow(f[N],Mod-2,Mod)

    for k in range(N-1,-1,-1):
        g[k]=((k+1)*g[k+1])%Mod
    return f,g

def Double_Factor(N):
    """ 0!!, 1!!, ..., N!! (mod Mod) を出力する.

    N: int
    """
    f=[1]*(N+1)
    for i in range(2,N+1):
        f[i]=i*f[i-2]%Mod
    return f

def Modular_Inverse(N):
    """ 1^(-1), 2^(-1), ..., N^(-1) (mod Mod) を出力する.

    [Input]
    N:int

    [Output]
    [-1, 1^(-1), 2^(-1), ..., N^(-1)] (第 0 要素に注意!!)
    """

    inv=[1]*(N+1); inv[0]=-1
    for k in range(2, N+1):
        q,r=divmod(Mod,k)
        inv[k]=(-q*inv[r])%Mod
    return inv

"""
組み合わせの数
Factor_with_inverse で fact, fact_inv を既に求めていることが前提 (グローバル変数)
"""

def nCr(n,r):
    """ nCr (1,2,...,n から相異なる r 個の整数を選ぶ方法) を求める.

    n,r: int
    """

    if 0<=r<=n:
        return fact[n]*(fact_inv[r]*fact_inv[n-r]%Mod)%Mod
    else:
        return 0

def nPr(n,r):
    """ nPr (1,2,...,n から相異なる r 個の整数を選び, 並べる方法) を求める.

    n,r: int
    """

    if 0<=r<=n:
        return (fact[n]*fact_inv[n-r])%Mod
    else:
        return 0

def nHr(n,r):
    """ nHr (1,2,...,n から重複を許して r 個の整数を選ぶ方法) を求める.

    n,r: int
    ※ fact, fact_inv は第 n+r-1 項まで必要
    """

    if n==r==0:
        return 1
    else:
        return nCr(n+r-1,r)

def Multinomial_Coefficient(*K):
    """ K=[k_0,...,k_{r-1}] に対して, k_0, ..., k_{r-1} に対する多項係数を求める.

    k_i: int
    """

    N=0
    g_inv=1
    for k in K:
        N+=k
        g_inv*=fact_inv[k]; g_inv%=Mod
    return (fact[N]*g_inv)%Mod

def Binomial_Coefficient_Modulo_List(n: int):
    """ n を固定し, r=0,1,...,n としたときの nCr (mod Mod) のリストを出力する.

    n: int

    [出力]
    [nC0 , nC1 ,..., nCn]
    """

    L=[1]*(n+1)
    inv=Modular_Inverse(n+1)
    for r in range(1, n+1):
        L[r]=((n+1-r)*inv[r]%Mod)*L[r-1]%Mod
    return L

def Pascal_Triangle(N: int):
    """
    0<=n<=N, 0<=r<=n の全てに対して nCr (mod M) のリストを出力する.

    N: int

    [出力]
    [[0C0], [1C0, 1C1], ... , [nC0, ... , nCn], ..., [NC0, ..., NCN]]
    """

    X=[1]
    L=[[1]]
    for n in range(N):
        Y=[1]
        for k in range(1,n+1):
            Y.append((X[k]+X[k-1])%Mod)
        Y.append(1)
        X=Y
        L.append(Y)
    return L

def Lucas_Combination(n, r):
    """ Lucas の定理を用いて nCr (mod Mod) を求める.

    """

    X=1
    while n or r:
        ni=n%Mod; ri=r%Mod
        n//=Mod; r//=Mod

        if ni<ri:
            return 0

        beta=fact_inv[ri]*fact_inv[ni-ri]%Mod
        X*=(fact[ni]*beta)%Mod
        X%=Mod
    return X

"""
等比数列
"""

def Geometric_Sequence(a, r, N):
    """ k=0,1,...,N に対する a*r^k を出力する.

    a,r,N: int
    """

    a%=Mod; r%=Mod
    X=[0]*(N+1); X[0]=a
    for k in range(1,N+1):
        X[k]=r*X[k-1]%Mod
    return X

def Geometric_Inverse_Sequence(a, r, N):
    """ k=0,1,...,N に対する a/r^k を出力する.

    a,r,N: int
    """

    a%=Mod; r_inv=pow(r, Mod-2, Mod)
    X=[0]*(N+1); X[0]=a
    for k in range(1,N+1):
        X[k]=r_inv*X[k-1]%Mod
    return X

"""
積和
"""
def Sum_of_Product(*X):
    """ 長さが等しいリスト X_1, X_2, ..., X_k に対して, sum(X_1[i]*X_2[i]*...*X_k[i]) を求める.
    """

    S=0
    for alpha in zip(*X):
        S+=product_modulo(*alpha)
    return S%Mod

def Sum_of_Product_Yielder(N,*Y):
    S=0
    M=len(Y)
    for _ in range(N+1):
        x=1
        for j in range(M):
            x*=next(Y[j]); x%=Mod
        S+=x
    return S%Mod
#==================================================
def solve():
    N,M=map(int,input().split())

    global Mod; Mod=998244353
    global fact, fact_inv

    fact,fact_inv=Factor_with_inverse(M)

    q,r=divmod(M,N)
    X=[q+1 if i<r else q for i in range(N)]
    return Multinomial_Coefficient(*X)

#==================================================
print(solve())
0