結果

問題 No.2136 Dice Calendar?
コンテスト
ユーザー googol_S0
提出日時 2022-11-25 21:44:55
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,429 bytes
コンパイル時間 385 ms
コンパイル使用メモリ 82,048 KB
実行使用メモリ 437,452 KB
最終ジャッジ日時 2024-10-02 04:22:17
合計ジャッジ時間 32,066 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 19 TLE * 2 -- * 3
権限があれば一括ダウンロードができます

ソースコード

diff # 

mod=998244353
def cmb(n,r):
  if r<0 or r>n:
    return 0
  return ((g1[n]*g2[r]%mod)*g2[n-r])%mod

N=300000
g1=[1]*(N+3)
for i in range(2,N+3):
  g1[i]=g1[i-1]*i%mod
g2=[0]*len(g1)
g2[-1]=pow(g1[-1],mod-2,mod)
for i in range(N+1,-1,-1):
  g2[i]=g2[i+1]*(i+1)%mod
inv=[0]*(N+3)
for i in range(1,N+3):
  inv[i]=g2[i]*g1[i-1]%mod

#X=[set() for i in range(21)]
P=[21**i for i in range(10)]
#X[0]=set([0])
#for i in range(20):
#  for j in X[i]:
#    for k in range(1,10):
#      X[i+1].add(j+P[k])
N=int(input())
S=[list(map(int,input().split())) for i in range(N)]
F=[(i in S[-1]) for i in range(10)]
X=[set() for i in range(N+1)]
X[0]=set([0])
for i in range(N-1):
  for j in X[i]:
    for k in range(6):
      X[i+1].add(j+P[S[i][k]])
ANS=0
for a in range(N+1):
  for b in range(a,N+1):
    for c in range(b,N+1):
      for d in range(c,N+1):
        for e in range(d,N+1):
          for f in range(e,N+1):
            for g in range(f,N+1):
              for h in range(g,N+1):
                i=N
                x=[0,a,b,c,d,e,f,g,h,i]
                j=0
                V=g1[N]
                for k in range(8,-1,-1):
                  x[k+1]-=x[k]
                  j+=P[k+1]*x[k+1]
                  if x[k+1]>1:
                    V=V*g2[x[k+1]]%mod
                for k in range(9):
                  if x[k+1]>0 and (j-P[k+1] in X[N-1]) and F[k+1]:
                    ANS+=V
                    break
print(ANS%mod)
0