368. Largest Divisible Subset -Medium

Question

Given a set of distinct positive integers, find the largest subset such that every pair (Si, Sj) of elements in this subset satisfies: Si % Sj = 0 or Sj % Si = 0.

If there are multiple solutions, return any subset is fine.

给出一个不重复的正整数集合，找出满足 Si % Sj = 0 或者 Sj % Si = 0 的最长可整除子序列。如果有多个解，只需返回任意一个即可

Example

nums: [1,2,3]

Result: [1,2] (of course, [1,3] will also be ok)

---

nums: [1,2,4,8]

Result: [1,2,4,8]

Solution

• 动态规划解。这道题和LIS非常相似，LIS的要求是递增，而最长可正整除序列的要求则是可整除。所以只要我们先将列表排序，这样只需判断 Si % Sj = 0 （i > j），再接下来就和LIS完全一样了（和我上一篇差不多就不写了）。不过这里需要输出一种结果。所以我们还需要额外保存每个元素的上一个元素索引。

  class Solution(object):
      def largestDivisibleSubset(self, nums):
          """
          :type nums: List[int]
          :rtype: List[int]
          """
          if len(nums) == 0: return []
          # 先排序保证只需要相除一次
          nums.sort()
          dp = [1] * len(nums)
          dp_index = [-1] * len(nums)  # 保存元素的上一个元素的索引，用于得到序列
          max_len = 1  # 维护一个最长子序列的长度
          for index_n, n in enumerate(nums):
              for i in range(index_n):
                  if n % nums[i] == 0 and dp[i] + 1 > dp[index_n]:
                      dp[index_n] = dp[i] + 1
                      dp_index[index_n] = i  # 每次更新dp时同时保存其上一个元素的索引
                      max_len = max(dp[i] + 1, max_len)
  
          result = []
          # 定位到第一个最长子序列的结尾
          index = dp.index(max_len)
          # 根据dp_index反向保存最长子序列
          while index != -1:
              result.append(nums[index])
              index = dp_index[index]  # 得到上一个元素的索引
          result.reverse()  # 倒序
          return result