diff --git a/ruby/198-House-Robber.rb b/ruby/198-House-Robber.rb new file mode 100644 index 000000000..bea75ac8d --- /dev/null +++ b/ruby/198-House-Robber.rb @@ -0,0 +1,11 @@ +# @param {Integer[]} nums +# @return {Integer} +def rob(nums) + dp = Array.new(nums.size + 3) { 0 } + + nums.each_with_index do |num, i| + dp[i+3] = num + [dp[i+1], dp[i]].max + end + + [dp[-1], dp[-2]].max +end