搜索(二)回溯

一、题目总结

基础问题

  • 46.全排列
  • 77.组合
  • 78.子集
  • 39.组合求和
  • 47.全排列 II(重复元素)
  • 90.子集 II(重复元素)
  • 40.组合总和II(重复元素)
  • 216.组合总和III
  • 113.路径总和 II

应用问题

  • 416.分割等和子集
  • 17.电话号码的字母组合
  • 131.分割回文串
  • 93.复原IP地址
  • 79.单词搜索
  • 51.N皇后(hard)
  • 37.解数独(hard)

二、题目

解决⼀个回溯问题,实际上就是⼀个决策树的遍历过程。过程大致是:

  • 找到问题的选择列表;
  • 选择当前的节点;
  • 往下一层继续选择;
  • 返回到上层的时候,会撤销对当前节点的选择。

代码框架:

DFS和回溯的区别:

  • DFS会对访问过的节点进行标记,表示以后不再重复访问该结点;
  • 而回溯则是选择当前的节点,往下一层继续选择;返回到上层的时候,会撤销对当前节点的选择,使其以后还可能被选择。

46.全排列

思路:套用回溯算法的模版,然后通过visited数组来排除在temp中已经选择过的数字。

回溯树:树中最底层的结点。

vector> ans;
vector temp;
void backtrack(vector& nums, vector visited){
    if (temp.size() == nums.size()) {
        ans.push_back(temp);
        return;
    }
    for (int i=0; i> permute(vector& nums) {
    vector visited(nums.size(), false);
    backtrack(nums, visited);
    return ans;
}

77.组合

思路:套用回溯算法的模版,然后通过传入一个 start 参数,来排除已经选择过的数字。

回溯树:树中第k层的结点。

vector> ans;
vector temp;
void backtrack(int n, int k, int start){
    if (temp.size() == k) {
        ans.push_back(temp);
    }
    for (int i=start; i<=n; i++) {
        temp.push_back(i);
        backtrack(n, k, i+1);
        temp.pop_back();
    }
}

vector> combine(int n, int k) {
    backtrack(n, k, 1);
    return ans;
}

78.子集

思路:套用回溯算法的模版,然后通过传入一个 start 参数,来排除已经选择过的数字。

回溯树:树中的所有结点。

vector> ans;
vector temp;
void backtrack(vector& nums, int start){
    ans.push_back(temp);
    for (int i=start; i> subsets(vector& nums) {
    backtrack(nums, 0);
    return ans;
}

39 组合总和

vector> ans;
vector temp;
void backtrack(vector& nums, int target, int start){
    if (target < 0) return;
    if (target == 0) ans.push_back(temp);
    
    for (int i=start; i> combinationSum(vector& candidates, int target) {
    backtrack(candidates, target, 0);
    return ans;
}

47.全排列 II

重复原因:有 2 个或以上个相同的元素在回溯树同一层被分别选择。-> 剪枝
剪枝操作:首先排序数组;然后与同一层的前一个进行比较。

在本题中,!visited[i-1]表示在同一层。

vector> ans;
vector temp;

void backtrack(vector& nums, vector visited){
    if (temp.size() == nums.size()) {
        ans.push_back(temp);
    }
    for (int i=0; i> permuteUnique(vector& nums) {
    sort(nums.begin(), nums.end());
    vector visited(nums.size(), false);
    backtrack(nums, visited);
    return ans;
}

90.子集 II

重复原因:有 2 个或以上个相同的元素在回溯树同一层被分别选择。-> 剪枝
剪枝操作:首先排序数组;然后与同一层的前一个进行比较。

在本题中,i>start表示在同一层。

vector> ans;
vector temp;
void backtracking(vector& nums, int start){
    ans.push_back(temp);
    
    for (int i=start; istart && nums[i] == nums[i-1]) {
            continue;
        }
        temp.push_back(nums[i]);
        backtracking(nums, i+1);
        temp.pop_back();
    }
}

vector> subsetsWithDup(vector& nums) {
    sort(nums.begin(), nums.end());
    backtracking(nums, 0);
    return ans;
}

40.组合总和II

同样,i>start表示在同一层。

vector> ans;
vector temp;
void backtrack(vector& nums, int target, int start){
    if (target < 0) return;
    if (target == 0) ans.push_back(temp);
    
    for (int i=start; istart && nums[i]==nums[i-1]) {
            continue;
        }
        temp.push_back(nums[i]);
        backtrack(nums, target-nums[i], i+1);
        temp.pop_back();
    }
}

vector> combinationSum2(vector& candidates, int target) {
    sort(candidates.begin(), candidates.end());
    backtrack(candidates, target, 0);
    return ans;
}

216.组合求和III

vector> ans;
vector temp;
void backtrack(int k, int target,int start){
    if (target < 0 || k < 0 ) return;
    if (target == 0 &&  k == 0) {
        ans.push_back(temp);
        return;
    }
    for (int i=start; i<10; i++) { // 选择列表
        temp.push_back(i); // 做选择
        backtrack(k-1, target-i, i+1); // 去一层选择
        temp.pop_back(); // 撤销选择
    }
}
vector> combinationSum3(int k, int target) {
    backtrack(k, target, 1);
    return ans;
}

113. 路径总和 II

二叉树中的路径回溯问题,还是那个套路。

vector> ans;
vector temp;
void backtrack(TreeNode* root, int sum){
    int v = root->val;
    temp.push_back(v);
    if (!root->left && !root->right && sum == v) {
        ans.push_back(temp);
    }else{
        if (root->left) backtrack(root->left, sum-v);
        if (root->right) backtrack(root->right, sum-v);
    }
    temp.pop_back();
}
vector> pathSum(TreeNode* root, int sum) {
    if (!root) return ans;
    backtrack(root, sum);
    return ans;
}

416.分割等和子集

思路:找出所有子集,如果有子集的和为数组和的一半,则返回true。

为了优化算法时间,对重复的子集进行剪枝。

bool ans = false;
void backtracking(vector& nums, int start, int sum, int target){
    if (ans) return;
    if (sum > target) return;
    
    if (sum == target) {
        ans = true;
        return;
    }
    
    for (int i=start; istart && nums[i]==nums[i-1]) {
            continue;
        }
        sum += nums[i];
        backtracking(nums, i+1, sum, target);
        sum -= nums[i];
    }
}

bool canPartition(vector& nums) {
    int s = accumulate(nums.begin(), nums.end(), 0);
    if (s % 2 != 0) return false;
    sort(nums.begin(), nums.end());
    backtracking(nums, 0, 0, s/2);
    
    return ans;
}

17.电话号码的字母组合

思路:

map mp={{'2',"abc"},{'3',"def"},{'4',"ghi"},{'5',"jkl"},  {'6',"mno"},{'7',"pqrs"},{'8',"tuv"},{'9',"wxyz"}};
vector ans;
string cur;

void backTracking(string digits, int idx){
    if (idx == digits.size()) {
        ans.push_back(cur);
        return;
    }
    
    string temp = mp[digits[idx]];
    for (int i=0; i letterCombinations(string digits) {
    if (digits.empty()) return ans;
    backTracking(digits, 0);
    return ans;
}

131.分割回文串

vector> ans;
vector temp;
bool check(string s){
    int l = 0, r = s.size()-1;
    while (l= s.size()) {
        ans.push_back(temp);
        return;
    }
    
    for (int i=start; i> partition(string s) {
    backtrack(s, 0);
    return ans;
}

93.复原IP地址

思路:

vector ans;
vector path;
bool isValid(string ip){
    if(stoi(ip)>255) return false;
    if(ip.size()>=2 && ip[0] == '0') return false;
    
    return true;
}
void backtracking(string s){
    if (path.size() == 4) {
        if (s.empty()) {
            string str = path[0] + '.' + path[1] + '.' + path[2] + '.' +path[3];
            ans.push_back(str);
        }
    }else{
        for (int i=1; i<=3 && i<=s.length(); i++) {
            string ip = s.substr(0, i);
            if (isValid(ip)) {
                path.push_back(ip);
                backtracking(s.substr(i, s.length()-i));
                path.pop_back();
            }
        }
    }
}
vector restoreIpAddresses(string s) {
    if (s.size() < 4) return ans;
    backtracking(s);
    return ans;
}

79.单词搜索

思路:深度优先搜索+简单的回溯

bool backtracking(vector>& board, string word, int i, int j, int k, vector>& visited){
    if (i<0 || i>=board.size() || j<0 || j>=board[0].size()) {
        return false;
    }
    if (board[i][j] != word[k] || visited[i][j]) {
        return false;
    }
    
    if (k == word.size()-1) {
        return true;
    }
    visited[i][j] = true;
    
    bool ans =  backtracking(board, word, i-1, j, k+1, visited) ||
    backtracking(board, word, i+1, j, k+1, visited) ||
    backtracking(board, word, i, j-1, k+1, visited) ||
    backtracking(board, word, i, j+1, k+1, visited);
    
    visited[i][j] = false;
    
    return ans;
}

bool exist(vector>& board, string word) {
    int m = board.size(), n = board[0].size();
    vector> visited(m, vector(n, false));
    
    for (int i=0; i

51.N皇后

问题描述:任意两个皇后都不能处于同一行、同一列或同一斜线上。

其实还是那个思路,并不难,一半的代码都在判断当前点能否放皇后。

vector> ans;
bool isValid(vector ans1, int n, int row, int col){
    // 行不用检查
    // 检查列
    for (int i=0; i=0 && j>=0); i--, j--) {
        if (ans1[i][j] != '.') return false;
    }
    // 检查右上
    for (int i=row-1, j=col+1; (i>=0 && j& board, int row){
    if (row == n) {
        ans.push_back(board);
        return;
    }
    for (int col=0; col> solveNQueens(int n) {
    vector board(n, string(n, '.'));
    backtrack(n, board, 0);
    return ans;
}

37. 解数独

问题描述:每一行、每一列、每一个粗线宫(3*3)内的数字均含1-9,不重复。给定数独永远是 9x9 形式的。

思路:下一层是下一列,如果列到尾了,就去下一行的第一个开始。

bool isValid(vector>& board, int r, int c, char ch){
    for (int i=0; i<9; i++) {
        if (board[r][i] == ch) return false;
        if (board[i][c] == ch) return false;
        if (board[r/3*3 + i/3][c/3*3 + i%3] == ch) return false;
    }
    return true;
}
bool backtrack(vector>& board, int i, int j){
    if (j == 9) return backtrack(board, i+1, 0); // 穷举到最后一列
    if (i == 9) return true; // 找到一个可行解
    
    if (board[i][j] != '.') {
         // 如果有预设数字,不用我们穷举
        return backtrack(board, i, j+1);
    }
    for (char ch = '1'; ch<='9'; ch++) {
        if (!isValid(board, i, j, ch)) continue;
        
        board[i][j] = ch;
        if (backtrack(board, i, j+1)) {
            // 如果找到一个可行解,立即结束
            return true;
        }
        board[i][j] = '.';
    }
    // 穷举完 1~9,依然没有找到可行解,此路不通,返回上一层。
    return false;
}
void solveSudoku(vector>& board) {
    backtrack(board, 0, 0);
}

你可能感兴趣的:(搜索(二)回溯)