g0101_0200.s0198_house_robber.solution.rb Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of leetcode-in-all Show documentation

104 LeetCode algorithm problem solutions

There is a newer version: 1.8

# #Medium #Top_100_Liked_Questions #Top_Interview_Questions #Array #Dynamic_Programming
# #Algorithm_I_Day_12_Dynamic_Programming #Dynamic_Programming_I_Day_3
# #Level_2_Day_12_Dynamic_Programming #Udemy_Dynamic_Programming #Big_O_Time_O(n)_Space_O(n)
# #2023_11_25_Time_41_ms_(99.28%)_Space_210.9_MB_(86.23%)

# @param {Integer[]} nums
# @return {Integer}
def rob(nums)
  return 0 if nums.empty?
  return nums[0] if nums.length == 1
  return [nums[0], nums[1]].max if nums.length == 2

  profit = Array.new(nums.length)
  profit[0] = nums[0]
  profit[1] = [nums[1], nums[0]].max

  (2...nums.length).each do |i|
    profit[i] = [profit[i - 1], nums[i] + profit[i - 2]].max
  end

  profit[nums.length - 1]
end