题目:
给定一个只包含正整数的非空数组。是否可以将这个数组分割成两个子集,使得两个子集的元素和相等。
注意:
- 每个数组中的元素不会超过 100
- 数组的大小不会超过 200
示例 1:
输入: [1, 5, 11, 5] 输出: true 解释: 数组可以分割成 [1, 5, 5] 和 [11].
首先可以看出如果数组和为奇数,是无法分割的(自然数乘以二都是偶数)。
然后其中一个基本想法是BFS,这个数字nums【i】我可以选择“加上”或者“不加”,
也就是双重递归。
思路如下:
def canPartition_(self, nums):
"""
:type nums: List[int]
:rtype: bool
"""
nums_sum=sum(nums)
if nums_sum%2!=0:
return False
else:
nums_sum//=2
rec=[False]
def search(nums,i,s=0):
if i==len(nums) or s>nums_sum:
return
if nums[i]+s==nums_sum:
rec[0]=True
return
else:
search(nums,i+1,s)
search(nums,i+1,s+nums[i])
search(nums,0,0)
return rec[0]
很可惜,这个方法只能AC不到一半的测试用例(可能C++或者Java会更多吧,毕竟没有递归限制)。
那么参考别人的答案后发现了用背包问题来解决的办法。
实际上我也想到过,但是无法转化成对应代码。
def canPartition(self,nums):
nums_sum = sum(nums)
if nums_sum % 2 != 0:
return False
else:
nums_sum //= 2
dp=[[0 for x in range(nums_sum+1)] for y in range(len(nums))]
for i in range(nums[0],len(dp[0])):
dp[0][i]=nums[0]
for i in range(1,len(nums)):
for j in range(nums[i],nums_sum+1):
dp[i][j]=max(dp[i-1][j],dp[i-1][j-nums[i]]+nums[i])
return dp[-1][-1]==nums_sum
然而这个代码没有通过,哈哈哈哈。
因为python太慢了,而我参考的java代码(写法几乎一样)是可以通过的。
对于这种可以转化为背包问题的问题,还需要多加练习。