[LeetCode]:动态规划+回溯 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]整除, 这就保证了状态转移方程的正确性. 

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

代码如下:

class Solution {
public:
    vector<int> largestDivisibleSubset(vector<int>& nums) {
    	vector<int> res;
    	if(nums.empty()) return res;
        vector<int> dp(nums.size(),1);
        vector<int> parent(nums.size(),0);

        sort(nums.begin(),nums.end(),greater<int>());

        for (int j = 0; j < nums.size()-1; ++j)
        {
        	for (int i = j+1; i < nums.size(); ++i)
        	{
        		if(nums[j]%nums[i]==0)
        		{
        			//dp[i]=max(dp[i],dp[j]+1);
        			//parent[i]=j;

        			if(dp[i]<dp[j]+1)
        			{
        				dp[i]=dp[j]+1;
        				parent[i]=j;
        			}

        		}
        	}
        }
        int maxval=0;
        int maxidx;
        for (int i = 0; i <nums.size() ; ++i)
        {
        	if(dp[i]>maxval)
        	{
        		maxval=dp[i];
        		maxidx=i;
        	}
        }
        

        for (int i = 0; i < maxval; ++i)
        {
        	res.push_back(nums[maxidx]);
        	maxidx=parent[maxidx];
        }

        return res;

    }
};

注意:不能图中注释的代码,此为错误代码。因为虽然dp的值未更新,但回溯的值更新了!

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值