一.题目:
最大可整除的子集合,这道题给了我们一个数组,让我们求这样一个子集合,集合中的任意两个数相互取余均为0.
Example 1:
Input: [1,2,3]
Output: [1,2] (of course, [1,3] will also be ok)
Example 2:
Input: [1,2,4,8]
Output: [1,2,4,8]
二.解题思路:
为了方便,我们可以先将集合从小到大排序成新数组nums,然后遍历数组的每个元素,扫描该元素之前的每个元素,根据动态规划的转移方程 dp[i] = dp[j] + 1 if (nums[i] % nums[j] == 0 and dp[i] < dp[j] +1) ,其中dp[i]表示数组nums[:i]中的LDS集合长度.
当完成了dp数组的求解后,我们并不能得到该最大长度对应的子集合内容,所以我们还需要设计一个数组parent,用来记录每次更新dp[i]时用到的j,这样当我们生成LDS内容时可以利用这些下标.
代码如下:
def largestDivisibleSubset(self, nums):
"""
:type nums: List[int]
:rtype: List[int]
"""
if not nums:return []
nums.sort()
N = len(nums)
dp, parents = [0]*N, [0]*N
mx ,mx_index = 0,-1
for i in range(N):
for j in range(i-1,-1,-1):
if nums[i] % nums[j] == 0 and dp[i] < dp[j] +1:
dp[i] = dp[j] + 1
parents[i] = j
if dp[i] > mx:
mx = dp[i]
mx_index = i
#dp[i]表示数组nums[:i]中的LDS集合长度
#所以这里mx+1才真正表示nums[:]中的LDS集合长度
result = []
for __ in range(mx+1):
result.append(nums[mx_index])
mx_index = parents[mx_index]
return result[::-1]
举例解释一下:
nums = [1,2,4,8,11,88]