5374. 长度为 n 的开心字符串中字典序第 k 小的字符串(回溯算法)

5374. 长度为 n 的开心字符串中字典序第 k 小的字符串

  • List res;	
    //答案集合不能定义为StringBuilder类型
    
  • 剩下的就是回溯算法

class Solution {
	char[] key = { 'a', 'b', 'c' };
    //res不能定义为StringBuilder类型
	List<String> res;

	public String getHappyString(int n, int k) {
		res = new LinkedList<String>();
        //暴力搜索构造字符串
		dfs(new StringBuilder(), n);
        //如果构造不出这么多字符串,返回空,也可以作为搜索的剪枝条件
		if (res.size() < k) {
			return "";
		} else {
			return res.get(k - 1);
		}
	}

	private void dfs(StringBuilder cur, int n) {
		if (cur.length() == n) {
			res.add(cur.toString());
			return;
		}
		if(cur.length() > n) {
			return;
		}
        //因为是按字典序添加的,也就不需要排序了
		for (int j = 0; j < key.length; j++) {
			if (cur.length() == 0 || cur.charAt(cur.length() - 1) != key[j]) {
				cur.append(key[j]);
				dfs(cur, n);
                //回溯
				cur.deleteCharAt(cur.length() - 1);
			}
		}
	}
}

你可能感兴趣的:(DFS,LeetCode)