hello-algo/ja/codes/ruby/chapter_dynamic_programming/climbing_stairs_dp.rb

=begin
File: climbing_stairs_dp.rb
Created Time: 2024-05-29
Author: Xuan Khoa Tu Nguyen ([email protected])
=end

### 階段の上り方：動的計画法 ###
def climbing_stairs_dp(n)
  return n  if n == 1 || n == 2

  # 部分問題の解を格納する dp テーブルを初期化する
  dp = Array.new(n + 1, 0)
  # 初期状態：最小の部分問題の解を設定する
  dp[1], dp[2] = 1, 2
  # 状態遷移：小さな部分問題から順に大きな部分問題を解く
  (3...(n + 1)).each { |i| dp[i] = dp[i - 1] + dp[i - 2] }

  dp[n]
end

### 階段の上り方：空間を最適化した動的計画法 ###
def climbing_stairs_dp_comp(n)
  return n if n == 1 || n == 2

  a, b = 1, 2
  (3...(n + 1)).each { a, b = b, a + b }

  b
end

### 実行用コード ###
if __FILE__ == $0
  n = 9

  res = climbing_stairs_dp(n)
  puts "#{n} 段の階段を上る方法は全部で #{res} 通り"

  res = climbing_stairs_dp_comp(n)
  puts "#{n} 段の階段を上る方法は全部で #{res} 通り"
end