字符串拼接 - 回溯

系列文章目录

---

前言

本人最近再练习算法，所以会发布自己的解题思路，希望大家多指教

一、题目描述

给定 M 个字符( a-z ) ，从中取出任意字符(每个字符只能用一次)拼接成长度为 N 的字符串，要求相同的字符不能相邻。

计算出给定的字符列表能拼接出多少种满足条件的字符串，输入非法或者无法拼接出满足条件的字符串则返回 0 。

二、输入描述

给定长度为 M 的字符列表和结果字符串的长度 N ，中间使用空格(" ")拼接。

0 < M < 30

0 < N ≤ 5

三、输出描述

输出满足条件的字符串个数。

输入：dde 2

输出：2
结果为：de,ed 两种子串

四、java代码

	//存放满足条件的所有子串
	static Set<String> strings = new HashSet<>();
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        String[] split = sc.nextLine().split(" ");
        //获取输入字符串
        String str = split[0];
        //获取子串长度
        int n = Integer.parseInt(split[1]);
        //通过深度搜索，生成满足要求的子串
        dfs(str,"", n);
        System.out.println(strings.size());
    }

    private static void dfs(String str, String prefix,  int n) {
        if(prefix.length() == n){
  			//当前子串符合要求，存入集合
            strings.add(prefix);
            return;
        }
        for (int i = 0; i < str.length(); i++) {
        	//当前字符如果不为第一个且已生成子串的最后一个字符和当前字符相同，则跳过
            if("" != prefix && prefix.substring(prefix.length()-1).equals(String.valueOf(str.charAt(i)))){
                continue;
            }
            //生成新的字符串
            String remaining = str.substring(0,i) + str.substring(i+1);
            dfs(remaining, prefix+str.charAt(i), n);
        }
    }

五、测试用例

输入：

java 3

输出：