LeetCode之回溯算法

什么是回溯算法：

对于回溯算法的定义，百度百科上是这样描述的：回溯算法实际上一个类似枚举的搜索尝试过程，主要是在搜索尝试过程中寻找问题的解，当发现已不满足求解条件时，就“回溯”返回，尝试别的路径。回溯法是一种选优搜索法，按选优条件向前搜索，以达到目标。但当探索到某一步时，发现原先选择并不优或达不到目标，就退回一步重新选择，这种走不通就退回再走的技术为回溯法，而满足回溯条件的某个状态的点称为“回溯点”。许多复杂的，规模较大的问题都可以使用回溯法，有“通用解题方法”的美称。

回溯算法其实就是一个不断探索尝试的过程，探索成功了也就成功了，探索失败了就在退一步，继续尝试……，

public void backtrack("原始参数") {
    //终止条件(递归必须要有终止条件)
    if ("终止条件") {
        //一些逻辑操作（可有可无，视情况而定）
        return;
    }

    for (int i = "for循环开始的参数"; i < "for循环结束的参数"; i++) {
        //一些逻辑操作（可有可无，视情况而定）

        //做出选择

        //递归
        backtrack("新的参数");
        //一些逻辑操作（可有可无，视情况而定）

        //撤销选择（防止污染分支）
    }
}

例如在LeetCode之N皇后问题上：

我们也是依次从第一行第一列开始一个一个遍历

先看看二叉树的先序遍历怎么写的：

public static void preOrder(TreeNode tree) {
    if (tree == null)
        return;
    visit(tree); //对当前结点操作
    preOrder(tree.left);
    preOrder(tree.right);
}

回溯算法相当于变为了N叉树的先序遍历，所以可以改为：

public static void preOrder(TreeNode tree) {
    if (tree == null)
        return;
    visit(tree) //对当前结点操作
    for (int i = 0; i <n ; i++) {
        preOrder("第i个子节点");
    }
}

所以N皇后的代码轮廓为：

public static void solveNQueens(char[][] chess, int row){ //row:行，col:列
    if("终止条件"){
        return;
    }
    for(int col = 0; col < chess[0].length; col++){
        //判断这个位置是否可以放皇后
        if(valid(chess, row, col){
            //如果可以放，我们就把原来的数组chess复制一份
            //因为数组是引用传递，如果直接在原数组上进行改变，会造成分支污染
            //注意！为什么是copy一份而不是直接int[][] temp = chess; ?
            //因为int[][] temp = chess;的话，对temp的更改会更改到chess上！
            int[][] temp = copy(chess);
            //然后在这个位置放上皇后
            temp[row][col] = 'Q';
            //下一行
            solveNQueens(temp, row + 1);
        }
    }
}

当然，为了减少空间的使用可以改为
public static void solveNQueens(char[][] chess, int row){ //row:行，col:列
    if("终止条件"){
        return;
    }
    for(int col = 0; col < chess[0].length; col++){
        //判断这个位置是否可以放皇后
        if(valid(chess, row, col){
            //然后在这个位置放上皇后
            chess[row][col] = 'Q';
            //下一行
            solveNQueens(temp, row + 1);
            //返回后，把当前点改回原样
            chess[row][col] = '.';
        }
    }
}