[leetcode] 368. Largest Divisible Subset

题目链接: https://leetcode.com/problems/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.

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]

思路: 可以用动态规划来解决. 为了使得问题可以转化为子问题, 最好将数组按照降序来排列, 然后当nums[j]%nums[i]==0的时候就可以得到一个状态转移方程dp[i] = max(dp[i], dp[j]+1), 因为数组按照降序排序, 所以nums[i] < nums[j],并且之前能够被nums[j]整除的数, 也必然能够别nums[i]整除, 这就保证了状态转移方程的正确性. 

他还要求找出最大结果, 所以我们还需要记录一下路径, 每一个数字, 我们记录一个第一个能够使其到达最大长度的父结点, 最后回溯一下即可.

代码如下:

[cpp]  view plain  copy

 

1. class Solution {  
2. public:  
3.     vector<int> largestDivisibleSubset(vector<int>& nums) {  
4.         if(nums.size() ==0) return {};   
5.         sort(nums.begin(), nums.end(), greater<int>());  
6.         int len = nums.size(), curMax=1, k=0;  
7.         vector<int> par(len), dp(len, 1), result;  
8.         for(int i =0; i < len; i++) par[i] = i;  
9.         for(int i =1; i < len; i++)  
10.         {  
11.             for(int j =0; j < i; j++)  
12.             {  
13.                 if(nums[j]%nums[i]!=0) continue;  
14.                 if(dp[i] < dp[j]+1) par[i] = j, dp[i]=dp[j]+1;  
15.                 if(dp[i] > curMax) curMax = dp[i], k = i;  
16.             }  
17.         }  
18.         while(par[k] != k)  
19.         {  
20.             result.push_back(nums[k]);  
21.             k = par[k];  
22.         }  
23.         result.push_back(nums[k]);  
24.         return result;  
25.     }  
26. };