結果
問題 |
No.2147 ハノイの塔のおもちゃ
|
ユーザー |
|
提出日時 | 2022-12-04 21:09:14 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 45 ms / 2,000 ms |
コード長 | 961 bytes |
コンパイル時間 | 437 ms |
コンパイル使用メモリ | 82,304 KB |
実行使用メモリ | 60,288 KB |
最終ジャッジ日時 | 2024-10-11 22:36:40 |
合計ジャッジ時間 | 1,551 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 8 |
ソースコード
import sys input = lambda: sys.stdin.readline().rstrip() MOD = 10**9+7 def main(): # 入力 N = int(input()) F = input().replace('A', '0').replace('B', '1').replace('C', '2') # 計算・出力 dp = [0, 0, 0] # i 個目までを A, B, C にまとめるのに必要な移動回数の最小 # j個目を X -> Y と動かすには j-1個目までを Z にまとめる必要がある # j段の塔を X -> Y と動かすには 2^n-1 回かかる # dp[j][k] = k = F[j]のとき # dp[j-1][k] # k != F[j]のとき # dp[j-1][!k and !F[j]] + 2^(j-1) pow2 = 1 for j in range(N): d = [0, 0, 0] fj = int(F[j]) for k in range(3): if k == fj: d[k] = dp[k] else: d[k] = (dp[3^k^fj] + pow2) % MOD dp = d pow2 *= 2 pow2 %= MOD print(dp[0]) if __name__ == "__main__": main()