搜索与图论 - DFS

1、题目描述

给定一个整数n,将数字1~n排成一排,将会有很多种排列方法。

现在,请你按照字典序将所有的排列方法输出。

输入格式

共一行,包含一个整数n。

输出格式

按字典序输出所有排列方案,每个方案占一行。

数据范围

1≤n≤71≤n≤7

输入样例:

3

输出样例:

1 2 3
1 3 2
2 1 3
2 3 1
3 1 2
3 2 1

2、分析

搜索与图论 - DFS_第1张图片

3、代码

idx 代表深搜的层数

idx = 0 代表根节点

idx = n 代表叶节点

import java.util.*;
import java.io.*;

public class Main{
    static int N = 10;
    static List> res = new ArrayList<>();
    static int n;
    
    public static void main(String[] args){
        Scanner in = new Scanner(new InputStreamReader(System.in));
        n = in.nextInt();
        
        StringBuilder sb = new StringBuilder();
        for(int i = 1;i <= n;i ++){
            sb.append(i);
        }
        
        boolean[] flag = new boolean[N];
        List list = new ArrayList<>();
        dfs(sb, list, flag, 0);
        
        for(int i = 0;i < res.size();i ++) {
            List tmp = res.get(i);
            for(String t : tmp){
                System.out.print(t + " ");
            }
            System.out.println();
        }
    }
    
    static void dfs(StringBuilder sb, List list, boolean[] flag, int idx){ //idx:层
        if(list.size() == n) { 
            // System.out.print("idx" + idx + " ");
            // System.out.print("list" + list.size() + " ");
            res.add(new ArrayList<>(list));
            return;
        }
        
        for(int i = 1;i <= sb.length();i ++) {
            if(!flag[i]) {
                list.add(String.valueOf(i));
                flag[i] = true;
                dfs(sb, list, flag, idx+1); //递归到下一层
                int size = list.size();
                list.remove(size - 1);
                flag[i] = false;
            }
            
        }
    }
}

 

你可能感兴趣的:(算法基础学习)