回溯法算法题

232 阅读1分钟

回溯法是暴力求解算法。 可适用于组合、切割、子集、排列、棋盘(n皇后)

如题1:

给定两个整数 n 和 k,返回 1 ... n 中所有可能的 k 个数的组合。

输入: n = 4, k = 2

输出:[[2,4],[3,4],[2,3],[1,2],[1,3],[1,4]]

var combine = function(n, k) {
    var temp=[];
    var res=[];
    function dfs(n,k,index){
        if(temp.length>=k){
            res.push([...temp]);
            return
        }
        for(var i=index;i<=n;i++){
            temp.push(i);
            dfs(n,k,i+1);
            temp.pop();//回退
        }
    }
    dfs(n,k,1);
    return res;
};

题2

给定一个 m x n 二维字符网格 board 和一个字符串单词 word 。如果 word 存在于网格中,返回 true ;否则,返回 false 。

单词必须按照字母顺序,通过相邻的单元格内的字母构成,其中“相邻”单元格是那些水平相邻或垂直相邻的单元格。同一个单元格内的字母不允许被重复使用。

image.png

示例1:

输入:board = [["A","B","C","E"],["S","F","C","S"],["A","D","E","E"]], word = "ABCCED"

输出:true

示例2:

输入:board = [["a","b"],["c","d"]], word = "abcd"

输出:false

var exist = function(board, word) {
    const r = board.length;
    const c = board[0].length;
   
    const wordIndexInit=0;
    for(let i=0;i<r;i++){
        for(let j=0;j<c;j++){
       if( dfs(board,word,i,j,wordIndexInit)){
           return true;
       }
    }
    }
    return false;
     function dfs(board,word,i,j,k){
        if(i<0||j<0||i>=r||j>=c||board[i][j]!==word[k]){
            return false;
        }
        if(k===word.length-1){
            return true
        }
        board[i][j]=''//标记已经查找过的
        let res=dfs(board,word,i+1,j,k+1)||dfs(board,word,i-1,j,k+1)
        ||dfs(board,word,i,j+1,k+1)||dfs(board,word,i,j-1,k+1)
        if(res===true){
            return true
        }else{
            board[i][j]=word[k]//回撤
        }
    }
};