leetcode 368.最大整除子集 - 动态规划
题干
给你一个由 无重复 正整数组成的集合 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 中的所有整数 互不相同
知识点&算法
整除子集要求集合中任意两个元素存在一方整除一方的关系。单独这么考虑或许有点难以入手,不妨对升序后的数组再进行考虑。
倘若已有整除子集{1,3},再加入的元素应当满足什么条件才能构成新的整除子集(数组递增且各元素不相同)?显然新加入的数应当是3的倍数。
对于原有的整除子集来说,只要新加入的数能被原来子集中的最大者整除,那么就一定能构成新的整除子集。
因此我们用dp[i]来维护以nums[i]为整除子集中最大元素的整除子集,当遍历到新的nums[i]时,在nums[i]左侧寻找dp[j]使得dp[j]的最后一位元素能整除nums[i],若在dp[j]的基础上插入nums[i]使得dp[i]的元素更多,则更新dp[i]。
最后返回元素最多的dp[i]即可。
题解
class Solution {
public:
vector<int> largestDivisibleSubset(vector<int>& nums) {
int n = nums.size(), resIdx = 0;
vector<vector<int> > dp(n+5);
sort(nums.begin(),nums.end());
for(int i = 0 ; i < n ; ++i) dp[i].push_back(nums[i]);
for(int i = 1 ; i < n ; ++i){
for(int j = 0 ; j < i ; ++j){
if(nums[i] % nums[j] == 0){
if(dp[i].size() < dp[j].size() + 1){
dp[i] = dp[j];
dp[i].push_back(nums[i]);
}
}
if(dp[i].size() > dp[resIdx].size() ){
resIdx = i;
}
}
}
return dp[resIdx];
}
};