LeetCode【每日一题】_491. 递增子序列_java

问题描述

491. 递增子序列

给定一个整型数组, 你的任务是找到所有该数组的递增子序列，递增子序列的长度至少是2。

示例:

输入: [4, 6, 7, 7]
输出: [[4, 6], [4, 7], [4, 6, 7], [4, 6, 7, 7], [6, 7], [6, 7, 7], [7,7], [4,7,7]]

说明:

1. 给定数组的长度不会超过15。
2. 数组中的整数范围是 [-100,100]。
3. 给定数组中可能包含重复数字，相等的数字应该被视为递增的一种情况。

题解

方法：递归枚举+剪枝

用递归枚举出所有的子序列，然后判断这个子序列是否满足下列情况：

• 长度>=2
• 递增
• 这个子序列没有出现过

递归枚举所有子序列的模板如下：

List<List<Integer>> ans = new ArrayList<List<Integer>>();

public void dfs(int[] nums, int i, ArrayList<Integer> list) {
	if (i == nums.length) {
		if (judge(list))
			//满足条件
			ans.add(new ArrayList<Integer>(list));
		return;
	}
	list.add(nums[i]);
	findSubsequences(nums, i + 1, nums[i], list);
	list.remove(list.size() - 1);
	findSubsequences(nums, i + 1, last, list);
}
//
public boolean judge(ArrayList<Integer> list) {
	//长度>=2,这个子序列没有出现过
	if (list.size() < 2 || ans.contains(list)) return false;
	//判断递增
	for (int i = 1; i < list.size(); i++) {
		if (list.get(i - 1) > list.get(i)) {
			return false;
		}
	}
	return true;
}

对于每一个元素，我们都有选择与不选择两条路

接下来就是考虑剪枝了，我们可以在选择之前加一个条件，判断当前元素是否大于等于当前子序列的最后一个元素，让递归枚举出的每一个子序列必定是递增的，这样就可以省去检查是否递增这一步了。

然后是去重，对于不选择这条路，当已经选择了某一元素时，我们判断后面如果还有这个元素，那就选择后面的，即当前元素不等于子序列最后一个元素时，我们的递归才继续下去，比如原序列[4, 6, 7, 7]，当子序列已经到[4, 6, 7]时，第一个7已经被选中，发现后面还有一个7，那就结束掉当前递归，回到上一步，不选择第一个7，这样便可实现去重

代码如下：

class Solution {
    List<List<Integer>> ans = new ArrayList<List<Integer>>();

	public List<List<Integer>> findSubsequences(int[] nums) {
		if (nums.length == 0)
			return new ArrayList<List<Integer>>();
		findSubsequences(nums, 0, Integer.MIN_VALUE, new ArrayList<Integer>());
		return ans;
	}

	private void findSubsequences(int[] nums, int i, int last, ArrayList<Integer> list) {
		if (i == nums.length) {
			if (list.size() >= 2)
				ans.add(new ArrayList<Integer>(list));
			return;
		}
		if (nums[i] >= last) {
			list.add(nums[i]);
			findSubsequences(nums, i + 1, nums[i], list);
			list.remove(list.size() - 1);
		}
		if (nums[i] != last)
			findSubsequences(nums, i + 1, last, list);
	}
}