华为机试：考古问题

【编程题目 | 200分】考古问题 [ 200 / 中等 ]

题目描述

考古问题，假设以前的石碑被打碎成了很多块，每块上面都有一个或若干个字符，请你写个程序来把之前石碑上文字可能的组合全部写出来，按升序进行排列。

示例

输入

3
a b c

输出

abc
acb
bac
bca
cab
cba

输入

3
a a b

输出

aab
aba
baa

思路分析

这道题是全排列问题，是含有重复元素的全排列的问题，按任意顺序返回所有不重复的全排列。

典型的回溯算法，回溯算法的重点是剪枝，在遍历的过程中，一边遍历一遍检测，在一定会产生重复结果集的地方剪枝。

如果要比较两个列表是否一样，一个容易想到的办法是对列表分别排序，然后逐个比对。既然要排序，我们就可以在搜索之前就对候选数组排序，一旦发现某个分支搜索下去可能搜索到重复的元素就停止搜索，这样结果集中不会包含重复列表。

可以参考：Java实现回溯算法入门（排列+组合+子集）

leetcode链接：47. 全排列 II

参考代码

import java.util.*;
import java.util.Scanner;

public class kaoGuProblem {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        int n = in.nextInt();
        in.nextLine();
        String[] arr = in.nextLine().split(" ");
        List<List<String>> res = new ArrayList<>();

        Arrays.sort(arr);  // 排序是剪枝的前提
        Deque<String> path = new ArrayDeque<>();  // 用栈去构造path,存储已经选了哪些数
        boolean[] used = new boolean[n];  // 用一个used数组判断是否被用过
        dfs(arr, 0, path, used, res); // 递归每一层
        for (int i = 0; i < res.size(); i++) {  // 输出结果
            System.out.println(String.join("", res.get(i)));
        }

    }
    public static void dfs(String[] arr, int depth, Deque<String> path, boolean[] used, List<List<String>> res) {
        if (depth == arr.length) {
            res.add(new ArrayList<>(path)); // 将path的拷贝加入到res中，就不会出现多余的空列表，要注意
            return;
        }
        for (int i = 0; i < arr.length; i++) {
            if (used[i]) {  // 如果使用过，则跳出
                continue;
            }
            if (i > 0 && arr[i].equals(arr[i - 1]) && !used[i - 1]) {
                continue;
            }
            path.addLast(arr[i]);
            used[i] = true;
            dfs(arr, depth + 1, path, used, res);
            path.removeLast();
            used[i] = false;
        }
    }
}

欢迎在评论区指正以及留下自己的更简洁的方法。