蚂蚁金服—JAVA实现全排列

最新推荐文章于 2023-02-20 17:49:17 发布

3分钟秒懂大数据

最新推荐文章于 2023-02-20 17:49:17 发布

阅读量962

点赞数 1

分类专栏： LeetCode 文章标签： java

本文链接：https://blog.csdn.net/weixin_38201936/article/details/124473893

版权

LeetCode 专栏收录该内容

7 篇文章 0 订阅

订阅专栏

该博客分享了一段蚂蚁金服的Java代码，用于实现数组的全排列。通过深度优先搜索（DFS）策略，该代码能够生成给定整数数组的所有可能排列组合，并将结果存储在List<List<Integer>>中。博客内容主要涉及数据结构与算法，特别是递归和回溯法的应用。

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

分享一道蚂蚁金服JAVA实现全排列的代码

在这里插入代码片
package com.lyz.dataStructure.LeetCode;

import java.util.List;
import java.util.ArrayDeque;
import java.util.ArrayList;
import java.util.Deque;

public class Solution{

    public static void main(String[] args) {
        int[] a = {1,2,3,4,5};
        List<List<Integer>>lists = permute(a);
        for(List<Integer> l:lists) {
            System.out.println(l);
        }
    }
    //对传入的数组进行全排列，将全排列的结果存放到List列表中，结果是多个list列表
    public static List<List<Integer>> permute(int[] nums){
        int len = nums.length;//len记录数组长度
        List<List<Integer>> res = new ArrayList<>();//存放全排列结果
        if(len==0) {
            return res;
        }
        Deque<Integer> path = new ArrayDeque<>();//用栈存储树的每一条路径，即全排列中的每种情况，Java中栈使用Deque
        boolean[] used = new boolean[len];//用used数组标记是否访问过
        dfs(nums,len,0,path,used,res);//从深度为0即根结点开始进行深度遍历
        return res;
    }
/*
 * nums数组是要求全排列的数组
 * len是数组长度，这个参数可写可不写，不写的话，函数中用到的时候要用nums.length
 * depth指排列树的深度，当深度与数组长度即元素个数相等时，说明找到了叶子结点，即答案中的一种情况
 * path用于记录树的每一条路径，即全排列中的每种情况
 * used数组标记每条路径上元素是否被访问过
 * res是要返回的最终结果
 */
    public static void dfs(int[] nums, int len, int depth, Deque<Integer> path, boolean[] used, List<List<Integer>> res) {
        if(len==depth) {//递归出口
            res.add(new ArrayList<>(path));//将path（全排列的一种情况）加入到结果列表中，注意这里加入的是拷贝！！！
            return;
        }
        for(int i=0;i<len;i++) {
            if(!used[i]) {//如果元素未被访问过，进行处理
                used[i]=true;//标记为已访问
                path.addLast(nums[i]);//由于是栈，所以将元素添加到栈的末尾
                dfs(nums, len, depth+1, path, used, res);//递归深搜
                //回溯
                used[i]=false;
                path.removeLast();//栈：后进先出
            }
        }
    }
}