回溯法解决的问题
回溯法,一般可以解决如下几种问题:
组合问题:N个数里面按一定规则找出k个数的集合
切割问题:一个字符串按一定规则有几种切割方式
子集问题:一个N个数的集合里有多少符合条件的子集
排列问题:N个数按一定规则全排列,有几种排列方式
棋盘问题:N皇后,解数独等等
组合问题
题目1:77. 组合
思路:循环套递归加回溯
1)递归法
class Solution {
List<List<Integer>> res=new ArrayList<>();
LinkedList<Integer> path=new LinkedList<>();
public List<List<Integer>> combine(int n, int k) {
backtracking(n,k,1);
return res;
}
//n是范围 k是k个数的组合 startIndex是控制取下一个节点继续组合
public void backtracking(int n,int k,int startIndex){
//收集结果
if(path.size()==k){
//说明拿到了一个个数为k的组合
res.add(new ArrayList<>(path));
}
//for循环控制节点不重复 取到的组合也不重复
for(int i=startIndex;i<=n;i++){
path.add(i);
//递归
backtracking(n,k,i+1);
//回溯
path.removeLast();
}
}
}
题目2:216. 组合总和 III
思路:回溯模版变种题
1)递归法
class Solution {
List<List<Integer>> result=new ArrayList<>();
LinkedList<Integer> path=new LinkedList<>();
public List<List<Integer>> combinationSum3(int k, int n) {
backtracking(k,n,0,1);
return result;
}
public void backtracking(int k,int n,int sum ,int startIndex){
//如果和为n 并且 结果是k个 收集结果
if(sum==n&&path.size()==k){
result.add(new ArrayList<>(path));
}
for(int i=startIndex;i<=9;i++){
path.add(i);
sum+=i;
backtracking(k,n,sum,i+1);
//回溯
sum-=i;
path.removeLast();
}
}
}
剪枝优化后的代码
class Solution {
List<List<Integer>> result=new ArrayList<>();
LinkedList<Integer> path=new LinkedList<>();
public List<List<Integer>> combinationSum3(int k, int n) {
backtracking(k,n,0,1);
return result;
}
public void backtracking(int k,int n,int sum ,int startIndex){
// 剪枝操作
if (sum > n) {
return;
}
// 剪枝操作 收集的元素比k大
if (path.size() > k) return;
//如果和为n 并且 结果是k个 收集结果
if(sum==n&&path.size()==k){
result.add(new ArrayList<>(path));
}
for(int i=startIndex;i<=9;i++){
path.add(i);
sum+=i;
backtracking(k,n,sum,i+1);
//回溯
sum-=i;
path.removeLast();
}
}
}
*题目3:17. 电话号码的字母组合
思路:得复习 还挺难理解
1)递归法
class Solution {
List<String> result=new ArrayList<>();
StringBuilder path=new StringBuilder();
public List<String> letterCombinations(String digits) {
if(digits==null || digits.length()==0){
return result;
}
String[] numString={"","","abc","def","ghi","jkl","mno","pqrs","tuv","wxyz"};
backtracking(digits,numString,0);
return result;
}
public void backtracking(String digits,String[] numString,int startIndex){
//收集结果
if(startIndex==digits.length()){
result.add(path.toString());
return;
}
//取出数字startIndex对应的字母
String str=numString[digits.charAt(startIndex)-'0'];
for(int i=0;i<str.length();i++){
path.append(str.charAt(i));
backtracking(digits,numString,startIndex+1);
path.deleteCharAt(path.length()-1);
}
}
}
题目4:39. 组合总和
思路:这个是可以重复选取元素的题 与前两题不同 就是一个递归参数的变化 不加一就可以
1)递归法
class Solution {
private List<List<Integer>> result=new ArrayList<>();
private LinkedList<Integer> path=new LinkedList<Integer>();
public List<List<Integer>> combinationSum(int[] candidates, int target) {
if(candidates.length==0){
return result;
}
bracktracking(candidates,target,0,0);
return result;
}
public void bracktracking(int[] candidates, int target,int sum,int startIndex){
if(sum>target){
return;
}
if(sum==target){
result.add(new ArrayList<>(path));
}
for(int i=startIndex;i<candidates.length;i++){
sum+=candidates[i];
path.add(candidates[i]);
//注意 此时i不+1
bracktracking(candidates,target,sum,i);
sum-=candidates[i];
path.removeLast();
}
}
}
*题目5:40. 组合总和 II
思路:这题有重复元素需先排序 其次要借助一下used数组来数层剪枝 有难度 需复习
1)递归法
class Solution {
private List<List<Integer>> result=new ArrayList<>();
private LinkedList<Integer> path=new LinkedList<Integer>();
public List<List<Integer>> combinationSum2(int[] candidates, int target) {
if(candidates.length==0){
return result;
}
//排序
Arrays.sort(candidates);
int[] used=new int[candidates.length];
bracktracking(candidates,target,0,0,used);
return result;
}
public void bracktracking(int[] candidates, int target,int sum,int startIndex,int[] used){
if(sum>target){
return;
}
if(sum==target){
result.add(new ArrayList<>(path));
}
for(int i=startIndex;i<candidates.length;i++){
//去重逻辑
if(i>0&&candidates[i]==candidates[i-1]&&used[i-1]==0){
continue;
}
sum+=candidates[i];
path.add(candidates[i]);
used[i]=1;
bracktracking(candidates,target,sum,i+1,used);
sum-=candidates[i];
path.removeLast();
used[i]=0;
}
}
}
切割问题
*题目6:131. 分割回文串
思路:因为切割过的地方,不能重复切割 和组合的回溯过程差不多 双指针法判断回文串
难点:
切割问题可以抽象为组合问题
如何模拟那些切割线
切割问题中递归如何终止
*在递归循环中如何截取子串
如何判断回文
1)递归法
class Solution {
List<List<String>> res=new ArrayList<>();
LinkedList<String> path=new LinkedList<>();
public List<List<String>> partition(String s) {
backtracking(s,0);
return res;
}
private void backtracking(String s,int startIndex){
//如果起始位置大于s的大小,说明找到了一组分割方案
if (startIndex >= s.length()) {
res.add(new ArrayList(path));
return;
}
for(int i=startIndex;i<s.length();i++){
//判断是否是回文串(难点)
if(checkPalindRome(s,startIndex,i)){
//切割子串(难点)
String str = s.substring(startIndex, i+1);
path.add(str);
} else {
continue;
}
backtracking(s, i + 1);
path.removeLast();
}
}
private boolean checkPalindRome(String s, int start, int end){
for (int i = start, j = end; i < j; i++, j--) {
if (s.charAt(i) != s.charAt(j)) {
return false;
}
}
return true;
}
}
**题目7:93. 复原 IP 地址
思路:
1)递归法
class Solution {
List<String> res = new ArrayList<>();
public List<String> restoreIpAddresses(String s) {
StringBuilder sb = new StringBuilder(s);
backTracking(sb, 0, 0);
return res;
}
private void backTracking(StringBuilder s, int startIndex, int dotCount){
if(dotCount==3){
if(isValid(s, startIndex, s.length() - 1)){
res.add(s.toString());
}
return;
}
for(int i = startIndex; i < s.length(); i++){
if(isValid(s, startIndex, i)){
//如果字符串合法拼上.
s.insert(i + 1, '.');
//递归i+2是因为i+1是.
backTracking(s, i + 2, dotCount + 1);
//回溯 把字符串还原
s.deleteCharAt(i + 1);
}else{
break;
}
}
}
//[start, end]
private boolean isValid(StringBuilder s, int start, int end){
if(start > end) return false;
// 0开头的数字不合法
if(s.charAt(start) == '0' && start != end) return false;
//当前字符串拼接的数
int num = 0;
for(int i = start; i <= end; i++){
//当前字符串对应的数字
int digit = s.charAt(i) - '0';
num = num * 10 + digit;
if(num > 255) return false;
}
return true;
}
}
子集问题
组合问题和分割问题收集结果的时候相当于收集的是部分 而子集收集的是全部 所以没有判断==k去收集结果
题目8:78. 子集
思路:
1)递归法
class Solution {
List<List<Integer>> result=new ArrayList<>();
LinkedList<Integer> path=new LinkedList<>();
public List<List<Integer>> subsets(int[] nums) {
if(nums.length==0){
return result;
}
bracktrack(nums,0);
return result;
}
void bracktrack(int[] nums,int startIndex){
result.add(new ArrayList<Integer>(path));
for(int i=startIndex;i<nums.length;i++){
path.add(nums[i]);
bracktrack(nums,i+1);
path.removeLast();
}
}
}
*题目9:90. 子集 II
思路:
1)递归法
class Solution {
List<List<Integer>> result =new ArrayList<>();
LinkedList<Integer> path=new LinkedList<>();
public List<List<Integer>> subsetsWithDup(int[] nums) {
if(nums.length==0){
return result;
}
int[] used=new int[nums.length];
Arrays.sort(nums);
bracktrack(nums,0,used);
return result;
}
void bracktrack(int[] nums,int startIndex,int[] used){
result.add(new ArrayList<Integer>(path));
for(int i=startIndex;i<nums.length;i++){
//树层去重 0就是false
if(i>0&&nums[i]==nums[i-1]&&used[i-1]==0){
continue;
}
path.add(nums[i]);
used[i]=1;
bracktrack(nums,i+1,used);
path.removeLast();
used[i]=0;
}
}
}
*题目10:491. 递增子序列
思路:
1)递归法
class Solution {
List<List<Integer>> res = new ArrayList<>();
List<Integer> path = new ArrayList<>();
public List<List<Integer>> findSubsequences(int[] nums) {
backTracking(nums, 0);
return res;
}
private void backTracking(int[] nums, int startIndex){
//题目要求至少有两个元素
if(path.size() >= 2) res.add(new ArrayList<>(path));
HashSet<Integer> hs = new HashSet<>();
for(int i = startIndex; i < nums.length; i++){
//如果最后一个元素大于此元素 就不能取了 或者是取到的元素在同一层已经用过了
if(!path.isEmpty() && path.get(path.size() -1 ) > nums[i] || hs.contains(nums[i]))
continue;
hs.add(nums[i]);
path.add(nums[i]);
backTracking(nums, i + 1);
path.remove(path.size() - 1);
}
}
}
排列问题
题目11:46. 全排列
思路:与以上问题不同的是排列问题不用startIndex参数了,因为排列问题,每次都要从头开始搜索,例如元素1在[1,2]中已经使用过了,但是在[2,1]中还要再使用一次1,而used数组,其实就是记录此时path里都有哪些元素使用了,一个排列里(递归里)一个元素只能使用一次
1)递归法
class Solution {
List<List<Integer>> result=new ArrayList<>();
LinkedList<Integer> path=new LinkedList<>();
public List<List<Integer>> permute(int[] nums) {
if(nums.length==0) return result;
boolean[] used=new boolean[nums.length];
backTrack(nums,used);
return result;
}
public void backTrack(int[] nums,boolean[] used){
if(path.size()==nums.length){
result.add(new ArrayList<Integer>(path));
return;
}
for(int i=0;i<nums.length;i++){
if(used[i]==true){
continue;
}
path.add(nums[i]);
used[i]=true;
backTrack(nums,used);
used[i]=false;
path.removeLast();
}
}
}
*题目12:47. 全排列 II
思路:used[i-1]==false是树层上去重,如果需要树枝上去重used[i-1]==true 是同一树枝 这题与子集II与组合总和 II题目 去重逻辑相同
1)递归法
class Solution {
List<List<Integer>> result=new ArrayList<>();
LinkedList<Integer> path=new LinkedList<>();
public List<List<Integer>> permuteUnique(int[] nums) {
if(nums.length==0){
return result;
}
boolean[] used=new boolean[nums.length];
Arrays.sort(nums);
backTrack(nums,used);
return result;
}
public void backTrack(int[]nums,boolean[] used){
if(nums.length==path.size()){
result.add(new ArrayList<Integer>(path));
return;
}
for(int i=0;i<nums.length;i++){
//去重逻辑
if(i>0&&nums[i]==nums[i-1]&&used[i-1]==false){
continue;
}
path.add(nums[i]);
used[i]=true;
backTrack(nums,used);
path.removeLast();
used[i]=false;
}
}
}
题目13:332. 重新安排行程
思路:
1)递归法
class Solution {
private LinkedList<String> res = new LinkedList<>();
private LinkedList<String> path = new LinkedList<>();
public List<String> findItinerary(List<List<String>> tickets) {
//给车票进行排序
Collections.sort(tickets, (a, b) -> a.get(1).compareTo(b.get(1)));
path.add("JFK");
//记录车票是否使用过
boolean[] used = new boolean[tickets.size()];
backTracking((ArrayList)tickets, used);
return res;
}
public boolean backTracking(ArrayList<List<String>> tickets, boolean[] used) {
// 机场的个数=航班数+1 证明找到了一条路径
if (path.size() == tickets.size() + 1) {
res = path;
return true;
}
for (int i = 0; i < tickets.size(); i++) {
//车票没用过并且车票地点与收集的路径到达地点相等 为了保证起始位置
if (!used[i] && tickets.get(i).get(0).equals(path.getLast())) {
//把到达地点添加到路径中
path.add(tickets.get(i).get(1));
//表示车票已经用过
used[i] = true;
if (backTracking(tickets, used)) {
return true;
}
used[i] = false;
path.removeLast();
}
}
return false;
}
}
棋盘问题
题目14:51. N 皇后
思路:
1)递归法
class Solution {
List<List<String>> res = new ArrayList<>();
public List<List<String>> solveNQueens(int n) {
char[][] chessboard = new char[n][n];
for (char[] c : chessboard) {
Arrays.fill(c, '.');
}
backTrack(n, 0, chessboard);
return res;
}
public void backTrack(int n, int row, char[][] chessboard) {
//如果递归的row 等于棋盘的行 就收集结果
if (row == n) {
List<String> list = new ArrayList<>();
for (char[] c : chessboard) {
list.add(String.copyValueOf(c));
}
res.add(list);
return;
}
//一个一个去试列
for (int col = 0;col < n; col++) {
if (isValid (row, col, n, chessboard)) {
chessboard[row][col] = 'Q';
//递归
backTrack(n, row+1, chessboard);
//回溯
chessboard[row][col] = '.';
}
}
}
public boolean isValid(int row, int col, int n, char[][] chessboard) {
// 检查列
for (int i=0; i<row; i++) { // 相当于剪枝
if (chessboard[i][col] == 'Q') {
return false;
}
}
// 检查45度对角线
for (int i=row-1, j=col-1; i>=0 && j>=0; i--, j--) {
if (chessboard[i][j] == 'Q') {
return false;
}
}
// 检查135度对角线
for (int i=row-1, j=col+1; i>=0 && j<=n-1; i--, j++) {
if (chessboard[i][j] == 'Q') {
return false;
}
}
return true;
}
}
题目15:37. 解数独
思路:
1)递归法
class Solution {
public void solveSudoku(char[][] board) {
solveSudokuHelper(board);
}
private boolean solveSudokuHelper(char[][] board){
//一个for循环遍历棋盘的行,一个for循环遍历棋盘的列,
// 一行一列确定下来之后,递归遍历这个位置放9个数字的可能性
for (int i = 0; i < 9; i++){ // 遍历行
for (int j = 0; j < 9; j++){ // 遍历列
if (board[i][j] != '.'){ // 跳过原始数字
continue;
}
// (i, j) 这个位置放k是否合适
for (char k = '1'; k <= '9'; k++){
if (isValidSudoku(i, j, k, board)){
board[i][j] = k;
if (solveSudokuHelper(board)){ // 如果找到合适一组立刻返回
return true;
}
board[i][j] = '.';
}
}
// 9个数都试完了,都不行,那么就返回false
return false;
}
}
// 遍历完没有返回false,说明找到了合适棋盘位置了
return true;
}
/**
* 判断棋盘是否合法有如下三个维度:
* 同行是否重复
* 同列是否重复
* 9宫格里是否重复
*/
private boolean isValidSudoku(int row, int col, char val, char[][] board){
// 同行是否重复
for (int i = 0; i < 9; i++){
if (board[row][i] == val){
return false;
}
}
// 同列是否重复
for (int j = 0; j < 9; j++){
if (board[j][col] == val){
return false;
}
}
// 9宫格里是否重复
int startRow = (row / 3) * 3;
int startCol = (col / 3) * 3;
for (int i = startRow; i < startRow + 3; i++){
for (int j = startCol; j < startCol + 3; j++){
if (board[i][j] == val){
return false;
}
}
}
return true;
}
}