【Lintcode】603. Largest Divisible Subset

题目地址：

https://www.lintcode.com/problem/largest-divisible-subset/description

给定一个数组，求其一个子集，使得每个数都是其后面那个数的约数。返回满足这样条件的最大的子集。

主要思路是动态规划。先将数组排序，接着设$f[i]$为以$nums[i]$结尾的最长上升子序列（这里的上升指的是前面的数整除后面的数）的长度，那么很显然$$f[i]=\underset{j<i\wedge nums[j]|nums[i]}{\max }(f[j])+1$$要得到方案的话，只需每次得到更长的子序列的时候，用一个数组记录这个子序列是由哪里得到的。最后再一路倒推回去即可。代码如下：

import java.util.Arrays;
import java.util.LinkedList;
import java.util.List;

public class Solution {
    /*
     * @param nums: a set of distinct positive integers
     * @return: the largest subset
     */
    public List<Integer> largestDivisibleSubset(int[] nums) {
        // write your code here
        // 之所以用链表，是因为最后得到答案的时候需要从头部插入数字
        List<Integer> res = new LinkedList<>();
        if (nums == null || nums.length == 0) {
            return res;
        }
        // 先排序
        Arrays.sort(nums);
        // dp[i]表示以nums[i]结尾的最长的满足条件子序列的长度
        int[] dp = new int[nums.length];
        dp[0] = 1;
        // prev[i]表示长度为dp[i]且以nums[i]结尾的满足条件的子序列中，nums[i]前一个数字的下标
        // 全初始化为-1，表示不存在前一个数字
        int[] prev = new int[nums.length];
        Arrays.fill(prev, -1);
        // maxLen表示最长序列的长度，longestLastIndex表示最长的序列的末尾数字的下标
        int maxLen = 0, longestLastIndex = 0;
        
        for (int i = 1; i < nums.length; i++) {
            for (int j = 0; j < i; j++) {
                if (nums[i] % nums[j] == 0 && dp[j] + 1 > dp[i]) {
                    dp[i] = dp[j] + 1;
                    prev[i] = j;
                }
            }
            if (dp[i] > maxLen) {
                maxLen = dp[i];
                longestLastIndex = i;
            }
        }
        
        while (longestLastIndex != -1) {
            res.add(0, nums[longestLastIndex]);
            // 不停的向前找序列中的上一个数字
            longestLastIndex = prev[longestLastIndex];
        }
        
        return res;
    }
}

时间复杂度$O(n^2)$，空间$O(n)$。