[leetcode] 368. Largest Divisible Subset

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.

Example 1:

nums: [1,2,3]

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

Example 2:

nums: [1,2,4,8]

Result: [1,2,4,8]

解法一：

整体的思路还是dp。由于是distinct数字，首先将数组从小到大排序，这样题中的条件就可以转换成后面一个数能不能被前面一个数整除。我们再定义一个dp数组，dp[i]表示从nums[i]到最后一个数字，最大可整除子集的长度。mx表示当前最大可整除子集的长度，mx_idx表示该子集的起始位置。parent数组来存储上一个可以整除nums[i]的数字的位置。

从后往前遍历数组，对每一个数字再遍历到末尾。如果nums[j]能整除nums[i]，而且dp[j] + 1>dp[i]，说明需要更新dp[i]的值，及其对应的parent，同时需要更新mx的值和mx_idx。

最后根据parent中数据，恢复出需要取出的数字。

class Solution {
public:
    vector<int> largestDivisibleSubset(vector<int>& nums) {
        sort(nums.begin(), nums.end());
        int mx = 0, mx_idx = 0;
        vector<int> dp(nums.size(),0), parent(nums.size(),0);
        
        for(int i=nums.size()-1; i>=0; i--){
            for(int j=i; j<nums.size();j++){
                if(nums[j]%nums[i]==0 && dp[i]<dp[j]+1){
                    dp[i] = dp[j]+1;
                    parent[i] = j;
                    if(mx<dp[i]){
                        mx = dp[i];
                        mx_idx = i;
                    }
                }
            }
        }
        
        vector<int> res;
        for(int i=0; i<mx; i++){
            res.push_back(nums[mx_idx]);
            mx_idx = parent[mx_idx];
        }
        return res;
        
    }
};