百度笔试（2021-9-7）子序列问题

算法 百度笔试（2021-9-7）子序列问题

@author：Jingdai
@date：2021.09.18

前几天做百度笔试的一个题没有a出来，今天有空，搜了大佬的题解，做了出来，现在总结一下。

题目描述

给你一个长度为 n 且全部由小写字母组成的字符串 s，问你有多少个子序列恰好包含 k 个字母？

输入描述，第一行两个整数 n 和 k，第二行输入一个长度为 n 的仅含小写字母的字符串 s。

输出描述，恰好包含k个字母的子序列的个数，由于个数可能会很大，所以输出对 1000000007 取余的结果。

示例输入1：

6 5
eecbad

示例输出1：

3

解释1：

显然有两个子序列"ecbad"满足要求，同时s自己也满足要求，因此答案为3

示例输入2：

10 2

示例输出2：

aaaccebecd

思路

我想估计很多人和我一样，首先想到的方法就是直接 dfs，对于字符串 s 的每个字母，都采取选和不选两种策略。在 dfs 的时候记录子序列的包含字母的个数，在叶子节点如果包含字母为 k，就使最后的结果加1。同时，当我们在 dfs 中如果我们发现当前包含的字母数已经超过 k 了，那么这个搜索树的子树已经不可能会是答案了，进行剪枝。最后写出的代码如下。

import java.util.*;

public class Main {
   

    private static int count = 0;

    public static void main(String[] args) {
   
        Scanner s = new Scanner(System.in);
        int n = s.nextInt();
        int k = s.nextInt();
        s.nextLine();
        char[] chars = s.nextLine().toCharArray();
        int[] memo = new int[26];

        dfs(chars, 0