結果

問題 No.3169 [Cherry 7th Tune] Desire for Approval
ユーザー 👑 Kazun
提出日時 2023-10-29 02:05:06
言語 PyPy3
(7.3.15)
結果
RE  
実行時間 -
コード長 7,256 bytes
コンパイル時間 228 ms
コンパイル使用メモリ 81,928 KB
実行使用メモリ 109,372 KB
最終ジャッジ日時 2025-05-30 21:03:13
合計ジャッジ時間 19,350 ms
ジャッジサーバーID
(参考情報)
judge1 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 17 RE * 29
権限があれば一括ダウンロードができます

ソースコード

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, mode=False):
    """
    0<=n<=N, 0<=r<=n の全てに対して nCr (mod M) のリストを出力する.

    N: int

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

    if mode:
        L=[[0]*(N+1) for _ in range(N+1)]
        L[0][0]=1
        for n in range(1,N+1):
            Ln=L[n]; Lnn=L[n-1]
            Ln[0]=1
            for r in range(1,N+1):
                Ln[r]=(Lnn[r]+Lnn[r-1])%Mod
        return L

    else:
        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 Catalan_Number(N):
    """ Catalan 数 C(N) を求める.

    注意
    C(N)=(2N)!/((N+1)!N!) なので, (2N)! までの値が必要.
    """

    g_inv=fact_inv[N+1]*fact_inv[N]%Mod
    return fact[2*N]*g_inv%Mod

"""
等比数列
"""

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 conv(P, Q):
    p_deg = len(P) - 1
    q_deg = len(Q) - 1
    r_deg = p_deg + q_deg

    R = [0] * (r_deg + 1)
    for i in range(p_deg + 1):
        for j in range(q_deg + 1):
            R[i + j] += P[i] * Q[j]
            R[i + j] %= Mod
    return R

class Exp_Poly:
    def __init__(self, b, poly):
        self.b = b
        self.poly = poly

    def __mul__(self, other):
        return Exp_Poly((self.b + other.b) % Mod, conv(self.poly, other.poly))

    def __repr__(self):
        return f"({self.b}, {self.poly})"

def expection(exp_poly):
    total = 0
    b = exp_poly.b
    poly = exp_poly.poly

    if b == 0:
        return 0

    b_inv = pow(b, Mod-2, Mod)
    b_inv_prod = b_inv
    for k, c in enumerate(poly):
        tmp = c * fact[k] % Mod
        total += tmp * (- b_inv_prod) % Mod; total %= Mod
        b_inv_prod *= b_inv; b_inv_prod %= Mod
    return total

def popcount(S):
    pop = 0
    while S:
        pop += 1
        x = S & (-S)
        S ^= x
    return pop

def solve():
    from itertools import product

    N = int(input())

    k = [0] * N; a = [0] * N
    for i in range(N):
        k[i], a[i] = map(int, input().split())

    global Mod, fact, fact_inv, inv
    Mod = 998244353
    fact, fact_inv = Factor_with_inverse(2000)
    inv = Modular_Inverse(2000)

    EP = [None] * N
    for i in range(N):
        EP[i] = Exp_Poly(pow(a[i], Mod - 2, Mod), [fact_inv[t] * pow(a[i], (Mod - 2) * t, Mod) % Mod for t in range(k[i])])

    prod = [None for _ in range(1 << N)]
    prod[0] = Exp_Poly(0, [1])

    D = [0] * (1 << N); D[0] = expection(prod[0])
    for S in range(1, 1 << N):
        x = S & (-S)
        i = x.bit_length() - 1
        prod[S] = prod[S ^ x] * EP[i]
        D[S] = expection(prod[S])

    E_pre = [0] * (N + 1)
    for V in product([0, 1, 2], repeat = N):
        S = T = U = 0
        for i in range(N):
            if V[i] == 0:
                S |= 1 << i
            elif V[i] == 1:
                T |= 1 << i
            else:
                U |= 1 << i

        deg = popcount(S | T)
        E_pre[deg] += pow(-1, popcount(T), Mod) * D[T | U]
        E_pre[deg] %= Mod

    E = [0] * (N + 1)
    E[N] = E_pre[N]
    for d in range(N-1, -1, -1):
        E[d] = (E[d + 1] + E_pre[d]) % Mod

    return E[1:]

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