LeetCode----House Robber

2034 단어 LeetCode동적 기획
House Robber
You are a professional robber planning to rob houses along a street. Each house has a certain amount of money stashed, the only constraint stopping you from robbing each of them is that adjacent houses have security system connected and it will automatically contact the police if two adjacent houses were broken into on the same night.
Given a list of non-negative integers representing the amount of money of each house, determine the maximum amount of money you can rob tonight without alerting the police.
Credits: Special thanks to @ifanchu for adding this problem and creating all test cases. Also thanks to @ts for adding additional test cases.
분석:
제목은 모든 원소의 비음수 그룹을 정하고, 그룹에서 원소를 선택하며,adjacent의 원소를 선택할 수 없는 상황에서 선택한 원소의 총계를 최대로 해야 한다는 뜻이다.
이것은 간단한 동적 기획 문제로 f[i]를 사용하여 아래에 i로 표시할 때 선택한 요소의 최대값을 표시하고nums는 주어진 수조이며 상태 이동 방정식은 다음과 같다.
f[i] = max{f[i - 1], f[i - 2] + nums[i]}
코드:
class Solution(object):
    def rob(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        f = [0] * 100  # record the max money
        nums_len = len(nums)
        if nums_len == 0:
            return 0
        elif nums_len == 1:
            return nums[0]
        elif nums_len == 2:
            return nums[0] if nums[0] > nums[1] else nums[1]
        else:
            f[0] = nums[0]
            f[1] = nums[0] if nums[0] > nums[1] else nums[1]
            for i in range(2, nums_len):
                f[i] = max(f[i - 1], f[i - 2] + nums[i])
            return f[nums_len - 1]

최적화 코드 1:
class Solution(object):
    def rob(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        nums_len = len(nums)
        it1, it2 = 0, 0
        for i in range(nums_len):
            it1 = it2 if it2 > (it1 + nums[i]) else (it1 + nums[i]) 
            it1, it2 = it2, it1
        return it2

좋은 웹페이지 즐겨찾기