问题1
你是一个专业的小偷,计划偷窃沿街的房屋。每间房内都藏有一定的现金,影响你偷窃的唯一制约因素就是相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警。
给定一个代表每个房屋存放金额的非负整数数组,计算你在不触动警报装置的情况下,能够偷窃到的最高金额。
示例 1:
输入: [1,2,3,1],输出: 4
解释: 偷窃 1 号房屋 (金额 = 1) ,然后偷窃 3 号房屋 (金额 = 3)。偷窃到的最高金额 = 1 + 3 = 4 。
示例 2:
输入: [2,7,9,3,1],输出: 12
解释: 偷窃 1 号房屋 (金额 = 2), 偷窃 3 号房屋 (金额 = 9),接着偷窃 5 号房屋 (金额 = 1)。偷窃到的最高金额 = 2 + 9 + 1 = 12 。
思路及解答
空间复杂度我有疑问?????见代码里的注释!!!
# 思路:动态规划。因为不能偷相邻的两家,所以当前最大的金钱数记为dp[i]=max(dp[i-1], dp[i-2]+money[i]) # 时间复杂度:O(N) # 空间复杂度:O(N) # 这里官方题解说是O(1),但我觉得应该是O(N)啊!??? class Solution: def rob(self, nums: List[int]) -> int: if nums == []: return 0 dp = [0, nums[0]] for i in range(2, len(nums)+1): dp.append(max(dp[i-1], dp[i-2]+nums[i-1])) # 因为dp补了dp[0]=0,因此nums与dp对应的索引应该有一个错位! return dp[-1]
问题2
你是一个专业的小偷,计划偷窃沿街的房屋,每间房内都藏有一定的现金。这个地方所有的房屋都围成一圈,这意味着第一个房屋和最后一个房屋是紧挨着的。同时,相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警。
给定一个代表每个房屋存放金额的非负整数数组,计算你在不触动警报装置的情况下,能够偷窃到的最高金额。
示例 1:
输入: [2,3,2],输出: 3
解释: 你不能先偷窃 1 号房屋(金额 = 2),然后偷窃 3 号房屋(金额 = 2), 因为他们是相邻的。
示例 2:
输入: [1,2,3,1],输出: 4
解释: 你可以先偷窃 1 号房屋(金额 = 1),然后偷窃 3 号房屋(金额 = 3)。偷窃到的最高金额 = 1 + 3 = 4 。
思路及代码
# 思路:第一个和最后一个不能同时偷,因此就转换成了 nums[0:len(nums)-1] 和 nums[1:len(nums)] 即nums[1:](去掉第一个), nums[:-1](去掉最后一个)基本的打家劫舍问题。 class Solution: def rob(self, nums: List[int]) -> int: if nums == []: return 0 if len(nums) == 1: return nums[0] # 待偷的序列:nums[0: len(nums)](即确定不偷最后一个) dp0 = [0, nums[0]] for i in range(2, len(nums)): dp0.append(max(dp0[i-1], dp0[i-2]+nums[i-1])) # 待偷的序列:nums[1: len(nums)+1](即确定不偷第一个,所以我们把第一个的数值置为0) dp1 = [0, 0, nums[1]] for j in range(3, len(nums)+1): dp1.append(max(dp1[j-1], dp1[j-2]+nums[j-1])) return max(dp0[-1], dp1[-1]) ############# 优化后的代码:把相同功能/结构的语句写成函数 ############## # Test_198 打家劫舍中的函数 def house_rob(nums): dp = [0, nums[0]] for i in range(2, len(nums)+1): dp.append(max(dp[i-1], dp[i-2]+nums[i-1])) return dp[-1] class Solution: def rob(self, nums: List[int]) -> int: if nums == []: return 0 if len(nums) == 1: return nums[0] return max(house_rob(nums[1:]), house_rob(nums[:-1]))
问题3
在上次打劫完一条街道之后和一圈房屋后,小偷又发现了一个新的可行窃的地区。这个地区只有一个入口,我们称之为“根”。 除了“根”之外,每栋房子有且只有一个“父“房子与之相连。一番侦察之后,聪明的小偷意识到“这个地方的所有房屋的排列类似于一棵二叉树”。 如果两个直接相连的房子在同一天晚上被打劫,房屋将自动报警。计算在不触动警报的情况下,小偷一晚能够盗取的最高金额。
示例 1:
输入: [3,2,3,null,3,null,1],输出: 7。
解释: 小偷一晚能够盗取的最高金额 = 3 + 3 + 1 = 7.
示例 2:
输入: [3,4,5,1,3,null,1],输出: 9。
解释: 小偷一晚能够盗取的最高金额 = 4 + 5 = 9.
思路及代码
小红书笔试题
思路及解答
# 思路:动态规划,第i个位置的点赞数,可以由第i-1位置并且不加第i位置的元素得到;也可以由第i-2个位置的点赞并且加上第i位置的点赞数得到(因为求的是点赞最大值,所以就是i-2的加当前的,这样才能保证最大) # 所以转移方程是:dp[i] = max(dp[i-1], dp[i+1]+data[i]),其中dp[i]表示[0:i+1]的点赞最大值,data[i]表示当前点赞数。 n = int(input()) a = list(map(int, input().split())) dp = [0, a[0]] for i in range(2, len(a)+1): dp.append(max(dp[i-1], dp[i-2]+a[i-1])) print(dp[-1])