回溯思想介绍

回溯法

回溯算法的基本思想是:从一条路往前走,能进则进,不能进则退回来,换一条路再试。这类思想常用于解决多方法问题(如组合问题之类的多种解的问题)。

回溯算法实际上就是在一棵树上进行深度优先遍历。而与深度优先搜索不同的是回溯:需要以要求的进行深度优先探索,必须每一步都符合要求。回溯法不能使用visited记录,因为同样的内容不同的序访问就会造成不同的结果,而不是仅仅“是否被访问过”这么简单。

要使用访问状态来记录,也就是对于每个点记录已经访问过的邻居方向,回溯之后从新的未访问过的方向去访问邻居。至于这点点之前有没有被访问过并不重要,重要的是没有以当前的序进行访问。

回溯指的是状态重置,因此递归之前做了什么,递归之后需要做相同操作的逆向操作,将其还原到原先的状态

组合问题

给定两个整数 nk,返回范围 [1, n] 中所有可能的 k 个数的组合。

你可以按 任何顺序 返回答案。

算法分析:该题是典型的组合问题,要找到所有满足条件的解,可以利用回溯思想。

例如n=4,k=2。即从[1, 4]中找到两数组合

如果组合里有 1 ,那么需要在 [2, 3, 4] 里再找 1 个数;
如果组合里有 2 ,那么需要在 [3, 4] 里再找 1数。注意:这里不能再考虑 1,因为包含 1 的组合,在第 1 种情况中已经包含。

设计说明:

设计一个列表用来临时存储排列组合,当组合数量满足题目要求,则将列表中的数存入目标容器。

  • 每一个结点递归地在做同样的事情,区别在于搜索起点,因此需要一个变量 Begin ,表示在区间 [Begin, n] 里选出若干个数的组合;
class Solution {
public:
    void dfs(int n, int k, int begin, vector<vector<int>> &V, vector<int> &path)
    {
        if(path.size()==k)//组合数量满足条件,将列表数存入容器
        {
            V.push_back(path);
            return ;
        }
        for(int i = begin;i<=n;i++)//n可以优化为n - (k - path.size()) + 1
        {
            path.push_back(i);//填入数
            dfs(n, k, i+1, V, path);// 下一轮搜索,设置的搜索起点要加 1,因为组合数不允许出现重复的元素
            path.pop_back();//深度优先遍历有回头的过程,因此递归之前做了什么,递归之后需要做相同操作的逆向操作
        }
    }

    vector<vector<int>> combine(int n, int k) 
    {
        vector<vector<int>> V; 
        if(n<k)
            return V;
        vector<int> path;//临时列表
        dfs(n, k, 1, V, path);
        return V;
    }
};

剪枝优化:我们可以观察出:当遍历起点从4开始,已不存在满足条件的组合,所以我们没必要继续遍历。

关键在于起点的上界,显然n并不是上界,因此我们需要找出上界

例如:n = 6 ,k = 4。

path.size() == 1 的时候,接下来要选择 3 个数,搜索起点最大是 4,最后一个被选的组合是 [4, 5, 6];
path.size() == 2 的时候,接下来要选择 2 个数,搜索起点最大是 5,最后一个被选的组合是 [5, 6];
path.size() == 3 的时候,接下来要选择 1 个数,搜索起点最大是 6,最后一个被选的组合是 [6];

可以归纳出:上界+接下来要选择的数-1=n

而接下来要选择的数 = k-path.size()

所以上界=n - (k - path.size()) + 1。

n皇后问题

在n*n的棋盘上放置n个皇后棋子,使其中任意两个皇后都不同列、不同行、不在一条斜线上。

算法策略:从第m行的第1列开始放置皇后,第m行放置后,再从第m+1行第1列开始放置皇后,若不能放置皇后则列数+1,直到能放皇后为止,若第n列都不能放皇后,则进行回溯(第m行列数+1再进行放置)。直到m = n时,将结果返回。

class Solution {
public:
    vector<vector<string>> solveNQueens(int n) {
        vector<vector<string>> result;
        vector<string> location(n, string(n, '.'));//记录每一行的结果
        generateBoard(0, n, result,  location);
        return result;
    }
		//判断该位置是否能够放置皇后
        int helper(int row, int col, vector<string> &location, int n)//row行,col列
        {
            int i, j;
            //检查是否同列
            for(i=0;i<row;i++)
            {
                if(location[i][col]=='Q')
                    return 0;
            }
            
			//检查是否同斜边
            for(i=x-1, j=y-1;i>=0&&j>=0;i--,j--)
            {
                if(location[i][j]=='Q')
                    return 0;
            }
            for(i=x-1, j=y+1;i>=0&&j<n;i--,j++)
            {
                if(location[i][j]=='Q')
                    return 0;
            }
            return 1;
            
        }

       void generateBoard(int k, int n, vector<vector<string>> &result,vector<string> &location)
       {
           if(n==k)
           {
               result.push_back(location);
                return ;
           }

           for(int i=0;i<n;i++)
           {
               //验证合法性
               if(helper(k, i, location, n))
               {
                   location[k][i] = 'Q';
                   generateBoard(k+1, n, result,  location);
                    location[k][i] = '.';//回溯
               }
           }
       }
};
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值