力扣---全排列---回溯

最新推荐文章于 2024-07-10 11:04:17 发布

sweet_Mary

最新推荐文章于 2024-07-10 11:04:17 发布

阅读量687

点赞数 27

分类专栏：力扣100 文章标签： leetcode 算法职场和发展 python

本文链接：https://blog.csdn.net/sweet_Mary/article/details/136943333

版权

力扣100 专栏收录该内容

29 篇文章 0 订阅

订阅专栏

本文介绍了使用递归和visit数组在C++和Python中实现整数数组排列的深度优先搜索（DFS）方法，强调了在将路径添加到结果列表时的深拷贝操作。

摘要由CSDN通过智能技术生成

思路：

递归做法，一般会有visit数组来判断第 i 位是否被考虑了。我们先考虑第0位，再考虑第1位，再考虑第2位...dfs函数中还是老套路，先判定特殊条件，再从当下的角度（决定第 j 位是哪个元素），依次遍历visit数组，依次将visit值为0的元素赋给第 j 位，并继续递归（记住要恢复现场）。

代码：

C++：

class Solution {
public:
    void dfs(vector<int>& visit,vector<vector<int>>& res,vector<int>& nums,vector<int>& path,int i,int len){
        if(i==len){
            res.push_back(path);
            return;
        }

        //考虑第i位
        for(int j=0;j<len;j++){
            if(visit[j]==0){
                visit[j]=1;
                path.push_back(nums[j]);
                dfs(visit,res,nums,path,i+1,len);
                visit[j]=0;
                path.pop_back();
            }
        }

    }

    vector<vector<int>> permute(vector<int>& nums) {
        vector<vector<int>> res;
        int len=nums.size();
        vector<int> visit(len,0);
        vector<int> path;
        dfs(visit,res,nums,path,0,len);
        return res;
    }
};

Python：

class Solution:
    def permute(self, nums: List[int]) -> List[List[int]]:
        def dfs(visit:List[int],res:List[List[int]],num:List[int],path:List[int],i:int,len_:int):
            if i==len_:
                res.append(path.copy())
                return 
            for j in range(len_):
                if visit[j]==0:
                    visit[j]=1
                    path.append(nums[j])
                    dfs(visit,res,nums,path,i+1,len_)
                    visit[j]=0
                    path.pop()
        res=[]
        len_=len(nums)
        visit=[0]*len_
        path=[]
        dfs(visit,res,nums,path,0,len_)
        return res

注意这句话：