leetcode 368.最大整除子集 - 动态规划

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];
    }
};