結果

問題 No.533 Mysterious Stairs
ユーザー lam6er
提出日時 2025-03-20 18:59:00
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 271 ms / 5,000 ms
コード長 735 bytes
コンパイル時間 223 ms
コンパイル使用メモリ 82,468 KB
実行使用メモリ 180,568 KB
最終ジャッジ日時 2025-03-20 18:59:59
合計ジャッジ時間 3,360 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

mod = 10**9 + 7
n = int(input())

if n == 0:
    print(0)
    exit()

# Initialize dp and sum_prev arrays
dp = [[0, 0, 0] for _ in range(n + 1)]
sum_prev = [0] * (n + 1)

for i in range(1, n + 1):
    for j_idx in range(3):
        j_val = j_idx + 1
        if j_val > i:
            dp[i][j_idx] = 0
            continue
        prev_i = i - j_val
        if prev_i >= 0:
            total_prev = sum_prev[prev_i]
            current_j_val = dp[prev_i][j_idx]
            sum_val = (total_prev - current_j_val) % mod
        else:
            sum_val = 0
        if prev_i == 0:
            sum_val = (sum_val + 1) % mod
        dp[i][j_idx] = sum_val
    sum_prev[i] = (dp[i][0] + dp[i][1] + dp[i][2]) % mod

print(sum_prev[n] % mod)
0