Leetcode回溯算法

回溯算法模板

List<value> result;
void backtrack(路径, 选择列表)
{
    if(满足约束条件)
    {
        reuslt.add(路径);
        return;
    }
    for(选择 : 选择列表)
    {
        //剪枝
        做选择；
        backtrack(路径，选择列表);
        撤销选择
    }
}

经典例题

全排列

题目

给定一个不含重复数字的数组 nums ，返回其 所有可能的全排列 。你可以 按任意顺序 返回答案。

输入：nums = [1,2,3]
输出：[[1,2,3],[1,3,2],[2,1,3],[2,3,1],[3,1,2],[3,2,1]]

回溯解法

class Solution {
public:
    vector<vector<int>> permute(vector<int>& nums) {
        //回溯
        vector<vector<int>> result;
        vector<int> path;
        backtrack(nums,path,result);
        return result;
    }
    void backtrack(vector<int>& nums,vector<int>& path,vector<vector<int>>& result)
    {
        if(path.size() == nums.size())    
        {
            result.push_back(path); 
            return;
        }   
        for(int i:nums)
        {
            if(find(path.begin(),path.end(),i)!=path.end()) continue;
            path.push_back(i);
            backtrack(nums,path,result);
            path.pop_back();
        }
    }
};

N皇后

题目

按照国际象棋的规则，皇后可以攻击与之处在同一行或同一列或同一斜线上的棋子。

n 皇后问题 研究的是如何将 n 个皇后放置在 n×n 的棋盘上，并且使皇后彼此之间不能相互攻击。

给你一个整数 n ，返回所有不同的 n 皇后问题 的解决方案。

每一种解法包含一个不同的 n 皇后问题 的棋子放置方案，该方案中 ‘Q’ 和 ‘.’ 分别代表了皇后和空位。

输入：n = 4
输出：[[".Q..","...Q","Q...","..Q."],["..Q.","Q...","...Q",".Q.."]]
解释：如上图所示，4 皇后问题存在两个不同的解法。

回溯解法

class Solution {
public:
    vector<vector<string>> solveNQueens(int n) {
        vector<string> path;
        vector<vector<string>> result;
        backtrack(0,n,path,result);
        return result;
    }
    void backtrack(int row,int N, vector<string>& path,vector<vector<string>>& result)
    {
        if(row == N)
        {
            result.push_back(path);
            return;
        }
        for(int i=0;i<N;i++)
        {
            if(!is_valid(row,i,path,N)) continue;
            string s(N,'.');
            s[i] = 'Q';
            path.push_back(s);
            backtrack(row+1,N,path,result);
            path.pop_back();
        }
    }
    bool is_valid(int row,int col,vector<string>& path,int N)
    {
        for(int i=0;i<row;i++)
        {
            if(path[i][col] == 'Q') return false;
            if(col-row+i>=0 && path[i][col-row+i] == 'Q')   return false;
            if(col+row-i<N && path[i][col+row-i] == 'Q')    return false;
        }
        return true;
    }
};