代码随想录算法训练营第二十六天|leetcode37、51题

一、leetcode第51题

本题将递归与回溯结合,重点在于如何判断该位置能否放置皇后,也就是该位置同一行、同一列、45度对角线和135度对角线是否已经放置了皇后,如果已经放置皇后则该位置不合法。因此,需要在每一行的每一列都试探性地插入皇后,如果合法则进行下一行的皇后插入,否则回溯倒退将皇后插入下一列。

具体代码如下:

cpp 复制代码
class Solution {
public:
    vector<vector<string>>result;
    vector<vector<string>> solveNQueens(int n) {
    vector<string>s(n,string(n,'.'));
    backtracking(n,0,s);
    return result;
    }
    void backtracking(int n,int row,vector<string>&s)
    {
       if(row==n)
       {
        result.push_back(s);
        return;
       }
       for(int col=0;col<n;col++)
       {
        if(isvalid(row,col,s,n))
        {
            s[row][col]='Q';
            backtracking(n,row+1,s);
            s[row][col]='.';
        }
       }
    }
    bool isvalid(int row,int col,vector<string>&s,int n)
    {
        for(int i=0;i<row;i++)
        {
            if(s[i][col]=='Q')
            {
                return false;
            }
        }
        for(int i=row-1,j=col-1;i>=0&&j>=0;i--,j--)
        {
            if(s[i][j]=='Q')
            {
                return false;
            }
        }
        for(int i=row-1,j=col+1;i>=0&&j<n;i--,j++)
        {
            if(s[i][j]=='Q')
            {
                return false;
            }
        }
        return true;
    }
};

二、leetcode第37题

本题比上一题要求要更为复杂,需要在每个位置都插入数字,因此需要进行双层回溯,对数独的每一个空位置都进行判断是否合法,如果合法则添加数字并继续探索下一个位置,如果不合法则需要回溯。

具体代码如下:

cpp 复制代码
class Solution {
public:
    void solveSudoku(vector<vector<char>>& board) {
    backtracking(board);
    }
    bool backtracking(vector<vector<char>>& board)
    {
        for(int i=0;i<board.size();i++)
        {
            for(int j=0;j<board[0].size();j++)
            {
                if(board[i][j]=='.')
                {
                    for(char k='1';k<='9';k++)
                    {
                        if(isvalid(i,j,board,k))
                        {
                            board[i][j]=k;
                            if(backtracking(board)){return true;}
                            board[i][j]='.';
                        }
                    }
                    return false;
                }
            }
        }
        return true;
    }
    bool isvalid(int i,int j,vector<vector<char>>& board,char k)
    {
        for(int a=0;a<9;a++)
        {
            if(board[i][a]==k)
            {
                return false;
            }
        }
        for(int b=0;b<9;b++)
        {
            if(board[b][j]==k)
            {
                return false;
            }
        }
        int starti=(i/3)*3;
        int startj=(j/3)*3;
        for(int i=starti;i<starti+3;i++)
        {
            for(int j=startj;j<startj+3;j++)
            {
                if(board[i][j]==k)
                {
                    return false;
                }
            }
        }
        return true;
    }
};
相关推荐
溟洵2 分钟前
【 C/C++ 算法】入门动态规划-----一维动态规划基础(以练代学式)
c语言·c++·算法
过河卒_zh15667667 分钟前
9.12AI简报丨腾讯投资AI游戏平台,B站开源AniSora V3
人工智能·算法·游戏·aigc·算法备案·生成合成类算法备案
boonya19 分钟前
Java垃圾回收机制理论算法及使用
jvm·算法·gc·垃圾收集器·理论
愚润求学19 分钟前
【贪心算法】day9
c++·算法·leetcode·贪心算法
lingran__1 小时前
速通ACM省铜第二天 赋源码(Adjacent XOR和Arboris Contractio)
c++·算法
半桔2 小时前
【STL源码剖析】二叉世界的平衡:从BST 到 AVL-tree 和 RB-tree 的插入逻辑
java·数据结构·c++·算法·set·map
塔中妖3 小时前
【华为OD】分割数组的最大差值
数据结构·算法·华为od
weixin_307779133 小时前
最小曲面问题的欧拉-拉格朗日方程 / 曲面极值问题的变分法推导
算法
RTC老炮3 小时前
webrtc弱网-AlrDetector类源码分析与算法原理
服务器·网络·算法·php·webrtc
孤廖3 小时前
【算法磨剑:用 C++ 思考的艺术・Dijkstra 实战】弱化版 vs 标准版模板,洛谷 P3371/P4779 双题精讲
java·开发语言·c++·程序人生·算法·贪心算法·启发式算法