搜索算法（三） 回溯法

1.回溯法

回溯法可以理解成一种特殊的深度优先算法，比起普通的DFS，多了还原当前节点的一步。

修改当前节点、递归子节点、还原当前节点。

本质是一种试错的思想。

维基百科：

 2.例题

1）

力扣https://leetcode.cn/problems/permutations/修改和还原当前节点放在了dfs函数的开头和结尾，中间遍历未访问的节点，递归访问这些节点。

class Solution {
public:
    vector<vector<int>> permute(vector<int>& nums) {
        int n = nums.size();
        vector<bool> visit(n,false);
        vector<int> ans;
        for(int i=0;i<n;i++){
            dfs(nums,visit,i,ans);
        }

        return res;
    }

    void dfs(vector<int>& nums, vector<bool>& visit, int k, vector<int>& ans){
        visit[k] = true;
        ans.push_back(nums[k]);
        int n = nums.size();
        if(ans.size()==n){
            res.push_back(ans);
        }else{
            for(int i=0;i<n;i++){
            if(!visit[i]){
                dfs(nums,visit,i,ans);
                }
            }
        }
        visit[k] = false;
        ans.pop_back();
    }

private:
    vector<vector<int>> res;    
};

2)

力扣https://leetcode.cn/problems/combinations/这题与上一题类似，只是停止搜索的条件变为组合长度为k

class Solution {
public:
    vector<vector<int>> combine(int n, int k) {
        vector<int> ans;
        for(int i=1;i<=n;i++){
            dfs(i,n,k,ans);
        }

        return res;
    }

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

private:
    vector<vector<int>> res;
};

3)

力扣https://leetcode.cn/problems/word-search/这题也是一样的回溯结构，深搜符合条件的路径

class Solution {
public:
    bool exist(vector<vector<char>>& board, string word) {
        int m = board.size();
        int n = board[0].size();
        vector<vector<bool>> visit(m,vector<bool>(n,false));
        bool res;
        for(int i=0;i<m;i++){
            for(int j=0;j<n;j++){
                if(board[i][j]==word[0]){
                    res = dfs(board,word,i,j,1,visit);
                    if(res==true){
                        return true;
                    }
                }
            }
        }
        return false;
    }

    bool dfs(vector<vector<char>>& board, string word, int a, int b, int x, vector<vector<bool>>& visit){
        if(word.size()==x) return true;
        visit[a][b] = true;
        int m = board.size();
        int n = board[0].size();
        for(int i=0;i<4;i++){
            int c = a + path[i];
            int d = b + path[i+1];
            if(c>=0 && c<m && d>=0 && d<n && visit[c][d]==false && 
                word[x]==board[c][d]){
                    if(dfs(board,word,c,d,x+1,visit)){
                        return true;
                    }
                }
        }
        visit[a][b] = false;
        return false;
    }
private:
    vector<int> path{-1,0,1,0,-1};
};

4)

力扣https://leetcode.cn/problems/n-queens/N皇后是经典的回溯问题，一行一行地放置皇后，每次放置前，判断一下是否符合每列每斜线只有有一个皇后的约束。放置皇后后，继续深搜下一行可能的放置方式，搜索完毕后，取消放置该行皇后。

class Solution {
public:
    vector<vector<string>> solveNQueens(int n) {
        vector<string> ans(n,string(n,'.'));
        dfs(0,n,ans);
        return res;
    }

    void dfs(int start, int n, vector<string>& ans){
       
        if(start==n){
            res.push_back(ans);
            return;
        }
        for(int i=0;i<n;i++){
             if(judge(ans,start,i)){
                ans[start][i] = 'Q';
                dfs(start+1, n, ans);
                ans[start][i] = '.';
             }
           
        }
    }

    bool judge(vector<string>& ans, int x, int y){
       int n = ans.size();
       for(int i=0;i<n;i++){
           if(ans[i][y]=='Q'){
               return false;
           }
       }
       for(int i=x-1, j=y-1;i>=0&&j>=0;i--,j--){
           if(ans[i][j]=='Q'){
               return false;
           }
       }
       for(int i=x-1,j=y+1;i>=0&&j<n;i--,j++){
           if(ans[i][j]=='Q'){
               return false;
           }
       }
       return true;
    }

private:
    vector<vector<string>> res;
};