码迷,mamicode.com
首页 > 其他好文 > 详细

【Leetcode】Combinations (Backtracking)

时间:2014-11-04 11:14:38      阅读:187      评论:0      收藏:0      [点我收藏+]

标签:style   io   color   ar   os   java   for   sp   on   

Given two integers n and k, return all possible combinations of k numbers out of 1 ... n.

For example,
If n = 4 and k = 2, a solution is:

[
  [2,4],
  [3,4],
  [2,3],
  [1,2],
  [1,3],
  [1,4],
]
这是一道经典的NP问题,采用回朔法

i从1开始,

先加入1,再加入2 temp=[1,2]

然后去掉2 加3 temp=[1,3]

然后去掉3 加4 temp=[1,4]

然后去掉4 i=5不合法了 结束for循环

然后去掉1

返回到i=1,然后i++


i现在是2了

先加入2,再加入3 temp=[2,3]

然后去掉3 加4 temp=[2,4]

然后去掉4 i=5不合法了 结束for循环

然后去掉2

返回到i=2,然后i++


i现在是3了

先加入3 再加入4 temp=[3,4]

然后去掉4 i=5不合法了 结束for循环

然后去掉3

返回到i=3,然后i++


i现在是4了 先加入4 i=5不合法了 结束for循环

然后去掉4

返回到4,然后i++


i现在是5了 结束for循环

全部结束

	public ArrayList<ArrayList<Integer>> combine(int n, int k) {
		ArrayList<ArrayList<Integer>> result = new ArrayList<ArrayList<Integer>>();
		ArrayList<Integer> temp = new ArrayList<Integer>();
		if (n <= 0 || k < 0)
			return result;
		helper(n, k, 1, temp, result);
		return result;
	}

	public void helper(int n, int k, int startNum, ArrayList<Integer> temp,
			ArrayList<ArrayList<Integer>> result) {
		if (k == temp.size()) {
			result.add(new ArrayList<Integer>(temp));
			return;
		}
		for (int i = startNum; i <= n; i++) {
			temp.add(i);
			helper(n, k, i + 1, temp, result);
			temp.remove(temp.size() - 1);
		}
	}
至于为什么要temp.remove?因为要保护现场,要知道return语句只能恢复i以前的值,却不能改变temp里面的值。





【Leetcode】Combinations (Backtracking)

标签:style   io   color   ar   os   java   for   sp   on   

原文地址:http://blog.csdn.net/yexiao123098/article/details/40779117

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!