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:
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]
这道题是找符合条件的最大集合,其中任意两个数字大的可以被小的整除,题目难度为Medium。
数组未排序,处理起来没有头绪,所以先把数组排序再做后续操作。
这里用sz[i]表示以第i个数字结尾的最大集合长度,例如集合{1, 2, 4, 8}对应的sz数组为{1, 2, 3, 4}。对于排序后数组中任意下标位置n对应的数字nums[n],以nums[n]结尾的最大集合要靠下标n之前的数字来确定,假设m为n之前的任意下标,如果nums[n]能够被nums[m]整除,同时sz[n] < sz[m]+1,表明将nums[n]加入以nums[m]结尾的集合后最大集合长度比当前记录的大,更新sz[n]为sz[m]+1,依此方法遍历nums[n]之前所有数字即可得到以nums[n]结尾的最大集合长度。这样遍历整个数组即可得出以任意位置数字结尾的最大集合长度,同时记录下最终所求的最大集合长度。
题目要求最大集合,而我们只记录了最大集合长度,如何得到最大集合呢?记录下以每个数字结尾的最大集合需要的额外存储空间略大,其实我们只需依次记录下最大集合所走过的路径即可逐步还原出最大集合,用数组preIdx记录每个位置数字加入最大集合前它的前一数字下标,这样通过类似链表的形式将最大集合存储了下来,最终依次将他们存入结果中。具体代码:
class Solution {
public:
vector<int> largestDivisibleSubset(vector<int>& nums) {
vector<int> sz(nums.size(), 0);
vector<int> preIdx(nums.size(), 0);
vector<int> ret;
int maxSz = 0, maxIdx = 0;
sort(nums.begin(), nums.end());
for(int i=0; i<nums.size(); ++i) {
for(int j=i; j>=0; --j) {
if(nums[i]%nums[j] == 0 && sz[i] < sz[j]+1) {
sz[i] = sz[j] + 1;
preIdx[i] = j;
if(maxSz < sz[i]) {
maxSz = sz[i];
maxIdx = i;
}
}
}
}
for(int i=0; i<maxSz; ++i) {
ret.push_back(nums[maxIdx]);
maxIdx = preIdx[maxIdx];
}
return ret;
}
};