結果

問題 No.2792 Security Cameras on Young Diagram
コンテスト
ユーザー Butterflv
提出日時 2024-06-21 22:22:39
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 638 bytes
コンパイル時間 369 ms
コンパイル使用メモリ 82,420 KB
実行使用メモリ 848,452 KB
最終ジャッジ日時 2024-06-21 22:22:44
合計ジャッジ時間 4,450 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other WA * 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)
0