[LeetCode] 368. Largest Divisible Subset @ python

一.题目:
最大可整除的子集合,这道题给了我们一个数组,让我们求这样一个子集合,集合中的任意两个数相互取余均为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]

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值