結果

問題 No.2792 Security Cameras on Young Diagram
ユーザー Butterflv
提出日時 2024-06-21 22:26:06
言語 PyPy3
(7.3.15)
結果
MLE  
実行時間 -
コード長 642 bytes
コンパイル時間 165 ms
コンパイル使用メモリ 82,412 KB
実行使用メモリ 848,552 KB
最終ジャッジ日時 2024-06-21 22:26:10
合計ジャッジ時間 4,036 ms
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 9 MLE * 1 -- * 10
権限があれば一括ダウンロードができます

ソースコード

diff #

from itertools import zip_longest
N=int(input())
A=tuple(map(int,input().split()))
MOD=998244353
dp=[[0 for i in range(A[0]+1)]]+[[0 for j in range(A[i]+1)]for i in range(N)]
dp[0][0] = 1
for i in range(len(dp)):
  for j in range(len(dp[i])):
    if i==0:
      dp[i][j] += dp[i][j-1]
    elif j==0:
      dp[i][j] += dp[i-1][j]
    else:
      dp[i][j] += dp[i-1][j] + dp[i][j-1]
    dp[i][j] %= MOD

ans = 0

dp=list(zip_longest(*dp))

for i in range(len(dp)):
  flag = False
  for j in range(len(dp[i])-1, -1, -1):
    if dp[i][j]!=None:
      if flag:
        ans += dp[i][j]
        break
      else:
        flag = True

print(ans%MOD)
0