LeetCode 0368 - Largest Divisible Subset

Reck Zhang

发布于 2021-08-11 11:32:40

2010

发布于 2021-08-11 11:32:40

文章被收录于专栏：Reck Zhang

Largest Divisible Subset

Desicription

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:

Input: [1,2,3]
Output: [1,2] (of course, [1,3] will also be ok)

Example 2:

Input: [1,2,4,8]
Output: [1,2,4,8]

Solution

class Solution {
public:
    std::vector<int> largestDivisibleSubset(std::vector<int>& nums) {
        std::sort(nums.begin(), nums.end());

        auto dp = std::vector<int>(nums.size(), 0);
        auto parent = std::vector<int>(nums.size(), 0);

        int maxNum = INT_MIN;
        int maxIndex = 0;

        for(int i = nums.size() - 1; i >= 0; i--) {
            for(int j = i; j < nums.size(); j++) {
                if(nums[j] % nums[i] == 0 && dp[i] < dp[j] + 1) {
                    dp[i] = dp[j] + 1;
                    parent[i] = j;
                }
            }

            if(dp[i] > maxNum) {
                maxNum = dp[i];
                maxIndex = i;
            }
        }

        auto res = std::vector<int>();
        for(int i = 0; i < maxNum; i++) {
            res.push_back(nums[maxIndex]);
            maxIndex = parent[maxIndex];
        }

        return res;
    }
};

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2019-06-30，如有侵权请联系 cloudcommunity@tencent.com 删除

distinct