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里面的值。
时间: 2024-10-06 07:07:14