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[i]表示以nums[i]最后一个元素的子集的长度，那么可以得到递推关系，对于j<i，如果nums[i]%nums[j]==0，dp[i]=max(dp[i].dp[j]+1)。这样得到了最大长度还不够，题目需要求出这个子集，所以还要维持一个数组记录每个数的上一个元素，并记录最终的最大子集的最后一个元素，这样才可以从它开始回溯。

class Solution {
public:
    vector<int> largestDivisibleSubset(vector<int>& nums) {
        int n=nums.size();
        if(n==0||n==1) return nums;
        sort(nums.begin(),nums.end());
        vector<int> dp(n,1);
        vector<int> parent(n,-1);
        int max_length=1;
        int max_index=-1;
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<i;j++)
            {
                if(nums[i]%nums[j]==0)
                {
                    if(dp[j]+1>dp[i]) 
                    {
                        parent[i]=j;
                        dp[i]=dp[j]+1;
                    }
                }
            }
            if(dp[i]>max_length)
            {
                max_length=dp[i];
                max_index=i;
            }
        }
        
        if(max_length==1) return vector<int>(1,nums[0]);
        vector<int> result;
        while(max_index!=-1)
        {
            result.push_back(nums[max_index]);
            max_index=parent[max_index];
        }
        reverse(result.begin(),result.end());
        return result;
    }
};