Scribbling

LeetCode: 198. House Robber 본문

Computer Science/Coding Test

LeetCode: 198. House Robber

focalpoint 2021. 12. 4. 16:56

DP 기초 문제로 맨날 보는 문제

class Solution:
    def rob(self, nums: List[int]) -> int:
        if len(nums) <= 2:
            return max(nums)
        dp = [0] * len(nums)
        dp[0] = nums[0]
        dp[1] = max(nums[0], nums[1])
        for i in range(2, len(nums)):
            dp[i] = max(dp[i-2]+nums[i], dp[i-1])
        return dp[len(nums)-1]