回溯法其实也是一种递归,本质上就是穷举,然后筛选出符合规则的数据。为了使回溯更加高效,我们根据规则要求,在穷举过程中加上条件限制(也就是剪枝)。
我们什么场景下应该想到使用回溯法呢?
如何画图去分析问题?
如何使用代码实现呢?
如何去优化程序?
组合是不强调元素顺序的,排列是强调元素顺序
面对回溯问题,我们第一想法就是画图!所有回溯法的问题都可以抽象为树形结构!
回溯法⼀般是在集合中递归搜索,集合的⼤⼩构成了树的宽度,递归的深度构成的树的深度。
public void Backtracking(参数) { if(满足终止条件) { 存放结果 return; } for(遍历本层集合中的元素) { 处理结点; dfs(参数);//递归 撤销处理该结点; //回溯 } }
看下面几个例子加深理解。
题目:给定两个整数 n 和 k,返回范围 [1, n] 中所有可能的 k 个数的组合。
你可以按 任何顺序 返回答案。
示例 1:
输入:n = 4, k = 2
输出:
[
[2,4],
[3,4],
[2,3],
[1,2],
[1,3],
[1,4],
]
我们已经知道组合是不考虑顺序的,[1,2],[2,1] 是一样的。所以我们在取数的时候,只需要取当前下标及后边下标即可。横向就是给定的n,纵向就是给定的k。按照示例n=4,k=2画出树状图如下:
//1.确定返回结果 IList<IList<int>> res = new List<IList<int>>();//存放符合条件的结果集合 List<int> path = new List<int>();//存放符合条件的结果 //2.确定入参 // * 题目给定的集合需要进入函数(此处是n,也可以是数组等集合) // * 需要知道终结条件,本题为k // * 题目要求返回组合,不考虑顺序,所以需要一个变量来记录当前选取n位置 //3.确定回溯方法 // startIndex用来记录本层递归的中,集合从哪里开始遍历 public void Backtracking(int n,int k,int startIndex) { //4.终止条件 if(path.Count==k) { //存放结果 res.Add(new List<int>(path)); return; } for(int i=startIndex;i<=n;i++) { path.Add(i);//处理节点 Backtracking(n,i+1);//递归 path.Remove(path.Count-1);//撤销处理该结点; 回溯 } }
按照上边,收集叶子节点就可以写出程序了。我们如何优化程序,如何实现剪枝呢?比如n=4,k=4,按照上图所示,会进入很多次不必要的分支。我们应该剪掉。当剩余节点不能够满足条件时,就不必继续进行了。如下图所示:
IList<IList<int>> res = new List<IList<int>>(); IList<int> path = new List<int>(); public IList<IList<int>> Combine(int n, int k) { BackTracking(n, k, 1); return res; } public void BackTracking(int n, int k, int startIndex) { if (path.Count == k) { res.Add(new List<int>(path)); return; } // n - (k - path.Count) + 1 //剪枝。如果for循环选择的起始位置之后的元素个数 已经不足 我们需要的元素个数了,那么就没有必要搜索了。 for (int i = startIndex; i <= n - (k - path.Count) + 1; i++) { path.Add(i); BackTracking(n, k, i + 1); path.RemoveAt(path.Count - 1); } }
题目:给定一个候选人编号的集合 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。
candidates 中的每个数字在每个组合中只能使用 一次 。
注意:解集不能包含重复的组合。
输入: candidates = [10,1,2,7,6,1,5], target = 8,
输出:
[
[1,1,6],
[1,2,5],
[1,7],
[2,6]
]
不包含重复的组合
例如当 {candidates} = [2, 2],{target} = 2时,上述算法会将列表 [2][2] 放入答案两次。
我们可以用一个Hash,咱.NET中使用Dictionary,将每次出现的元素和频率记录下来。不过更方便的是我们将candidates排序,查看是否和上一个元素相同。
private IList<IList<int>> result = new List<IList<int>>();//结果集 private List<int> path = new List<int>();//结果 int sum; /// <summary> /// /// </summary> /// <param name="candidates">元素数组</param> /// <param name="startIndex">开始下标</param> /// <param name="isUsed">记录是否使用</param> /// <param name="target">条件</param> public void BackTracking(int[] candidates, int startIndex, bool[] isUsed, int target) { if (sum == target) //终止条件 { result.Add(new List<int>(path));//收集结果 return; } for (int i = startIndex; i < candidates.Length && candidates[i] + sum <= target; i++) { //出现重复节点,同层的第一个节点已经被访问过,所以直接跳过 if (i > 0 && candidates[i] == candidates[i - 1] && !isUsed[i - 1]) { continue; } isUsed[i] = true; sum += candidates[i]; path.Add(candidates[i]); //每个节点仅能选择一次,所以从下一位开始 BackTracking(candidates, i + 1, isUsed, target); int temp = path[path.Count - 1]; path.Remove(path[path.Count - 1]); isUsed[i] = false; sum -= temp; } }
candidates[i] + sum <= target
if (i > 0 && candidates[i] == candidates[i - 1] && !isUsed[i - 1])
private IList<IList<int>> result=new List<IList<int>>(); private IList<int> path=new List<int>(); int sum; public IList<IList<int>> CombinationSum2(int[] candidates, int target) { Array.Sort(candidates); bool[] used = new bool[candidates.Length]; BackTracking(candidates, 0, used, target); return result; } public void BackTracking(int[] candidates, int index, bool[] used, int target) { if (sum == target) { result.Add(new List<int>(path)); return; } for (int i = index; i < candidates.Length && candidates[i]+sum<=target; i++) { //出现重复节点,同层的第一个节点已经被访问过,所以直接跳过 if (i > 0 && candidates[i] == candidates[i - 1] && !used[i - 1]) { continue; } used[i] = true; sum += candidates[i]; path.Add(candidates[i]); //每个节点仅能选择一次,所以从下一位开始 BackTracking(candidates, i+1, used, target); int temp = path[path.Count - 1]; path.Remove(path[path.Count-1]); used[i] = false; sum -= temp; } }
题目:找出所有相加之和为 n 的 k 个数的组合,且满足下列条件:
只使用数字1到9
每个数字 最多使用一次
返回 所有可能的有效组合的列表 。该列表不能包含相同的组合两次,组合可以以任何顺序返回。
示例 1:
输入: k = 3, n = 7
输出: [[1,2,4]]
解释:
1 + 2 + 4 = 7
其实这道题和第一题差不多,就是收集结果时加了一个和的限制。
IList<IList<int>> res = new List<IList<int>>(); IList<int> path = new List<int>(); /// <summary> /// /// </summary> /// <param name="targetSum">给定目标和</param> /// <param name="k">要求元素个数</param> /// <param name="startIndex">开始下标</param> /// <param name="sum">当前已选取元素和</param> public void BackTracking(int targetSum, int k, int startIndex, int sum) { // 减枝 if (sum > targetSum) { return; } if (path.Count == k) { if (targetSum == sum) res.Add(new List<int>(path)); return; } // 减枝 9 - (k - path.size()) + 1 for (int i = startIndex; i <= 9 - (k - path.Count) + 1; i++) { path.Add(i); sum += i; BackTracking(targetSum, k, i + 1, sum); //回溯 path.RemoveAt(path.Count - 1); //回溯 sum -= i; } }
IList<IList<int>> res = new List<IList<int>>(); IList<int> path = new List<int>(); public IList<IList<int>> CombinationSum3(int k, int n) { BackTracking(n, k, 1, 0); return res; } public void BackTracking(int targetSum, int k, int startIndex, int sum) { // 减枝 if (sum > targetSum) { return; } if (path.Count == k) { if (targetSum == sum) res.Add(new List<int>(path)); return; } // 减枝 9 - (k - path.size()) + 1 for (int i = startIndex; i <= 9 - (k - path.Count) + 1; i++) { path.Add(i); sum += i; BackTracking(targetSum, k, i + 1, sum); //回溯 path.RemoveAt(path.Count - 1); //回溯 sum -= i; } }
组合是无序的,所以需要strartIndex来控制选择范围,查找当前结果时,不可以再次使用。当然如果选取元素不是同一集合,就不要用startIndex了。比如17. 电话号码的字母组合
题目给的元素集合,是否存在重复元素,如果存在,需要对同层相同已使用元素进行剪枝
结果长度是否有要求?如果有要求判断剩余元素是否满足长度要求。