给你一个由 无重复 正整数组成的集合 nums ,请你找出并返回其中最大的整除子集 answer ,子集中每一元素对 (answer[i], answer[j]) 都应当满足:
answer[i] % answer[j] == 0 ,或
answer[j] % answer[i] == 0
如果存在多个有效解子集,返回其中任何一个均可。
示例 1:
输入:nums = [1,2,3]
输出:[1,2]
解释:[1,3] 也会被视为正确答案。
示例 2:
输入:nums = [1,2,4,8]
输出:[1,2,4,8]
提示:
1 <= nums.length <= 1000
1 <= nums[i] <= 2 * 109
nums 中的所有整数 互不相同
思路:动态规划。经过分析,我们排序后,只需要求相邻的两个元素是不是能整除即可。因此用f[i]表示前i个元素所能构成的最大整除子集,然后记录最大的位置。最后倒着推一遍,就能得到DP的方案。
class Solution {
public:
vector<int> largestDivisibleSubset(vector<int>& nums) {
int n = nums.size();
sort(nums.begin(), nums.end());
vector<int> f(n);
int k = 0;
for(int i = 0; i < n; i ++ ) {
f[i] = 1;
for(int j = 0; j < i; j ++ ) {
if(nums[i] % nums[j] == 0) {
f[i] = max(f[i], f[j] + 1);
if(f[k] < f[i]) k = i;
}
}
}
vector<int> res;
res.push_back(nums[k]);
while(f[k] > 1) {
for(int i = 0; i < k; i ++ ) {
if(nums[k] % nums[i] == 0 && f[k] == f[i] + 1) {
res.push_back(nums[i]);
k = i;
break;
}
}
}
return res;
}
};