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

思路

• 又又又又又是 dp ! 首先对 nums数组从小到大排序，复杂度为O(NlogN）。
• 建立一维 dp 数组，长度为 nums.length。其中 dp[i] 表示以 nums[i] 为最大元素的子集，能包含的元素个数最多为多少。
• 边界条件：由于 dp 数组的定义必须包含当前位置的元素，所以 dp[i] 初始化为 1。
• 状态转移。对于任意一个元素 nums[i]，如果在 [0,i-1]元素中有一个子集，且 nums[i] 能够整除该子集中的最大元素，则将 nums[i] 加入该子集可以得到一个新子集，该子集以 nums[i] 为最大元素。 遍历所有的以 nums[0,…,i-1] 为最大元素的子集，取元素个数最多的加入 nums[i]，构成以 nums[i] 为最大元素且包含元素数量最多的子集。 具体的，dp[i] = max { dp[i] , d[j] + 1 },其中 j ∈ [0,i-1]。
• 还有一个问题就是dp完了之后如何构建结果子集？这里说明一下构建方法。 1、首先在 dp 的时候记录最大子集大小为 maxSize，其中最大元素为 x。 2、将 x 加入结果列表，同时 maxSize -= 1。 3、接着在 x 元素之前，查找满足 dp[i] = maxSize，且 x % nums[i] == 0的元素，该元素就是 x 的前驱元素，加入到结果列表。 4、更新 x 为 找到的 nums[i]，同时maxSize -= 1，继续第3步，直到 maxSize == 0

dp 操作有两层循环，时间复杂度为 O(N^2)

代码

class Solution {
    public List<Integer> largestDivisibleSubset(int[] nums) {
    
    	int len = nums.length;
    	Arrays.sort(nums);
    	
    	//dp[i]表示以i为最大元素的子集中，元素的个数最多为多少
    	int []dp = new int[len];
    	Arrays.fill(dp,1);
    	//记录最大子集的元素个数，以及最大元素的位置
    	int maxSize = 1;
    	int flag = 0;
    	
    	//遍历i，对于每个i，要考虑前面所有的j
    	for(int i = 0;i < len;i++) {
    		for(int j = 0;j < i;j++) {
    			//如果可以整除，则表示可以把nums[i]接在后面构成新的子集
    			if(nums[i] % nums[j] == 0) {
    				dp[i] = Math.max(dp[i], dp[j] + 1);
    				if(dp[i] > maxSize) {
    					maxSize = dp[i];
    					flag = i;
    				}
    			}
    		}
    	}
    	
    	//dp完毕后构建目标子集输出
    	List<Integer> lst = new ArrayList<Integer>();
    	lst.add(nums[flag]);
    	maxSize -= 1;
    	int lastNumber = nums[flag];
    	for(int i = flag-1;i >= 0; i--) {
    		if(dp[i] == maxSize && lastNumber % nums[i] == 0) {
    			lst.add(nums[i]);
    			maxSize -= 1;
    			lastNumber = nums[i];
    			if(maxSize == 0)
    				break;
    		}
    	}
    	return lst;
    }
}