LCP 25-古董键盘

小扣在秋日市集购买了一个古董键盘。由于古董键盘年久失修，键盘上只有 26 个字母 a~z 可以按下，且每个字母最多仅能被按 k 次。
小扣随机按了 n 次按键，请返回小扣总共有可能按出多少种内容。由于数字较大，最终答案需要对 1000000007 (1e9 + 7) 取模。 示例
1： >输入：k = 1, n = 1 > >输出：26 > >解释：由于只能按一次按键，所有可能的字符串为 “a”, “b”, … “z”
示例 2： >输入：k = 1, n = 2 > >输出：650 >

解释：由于只能按两次按键，且每个键最多只能按一次，所有可能的字符串（按字典序排序）为 “ab”, “ac”, … “zy” 提示： - 1 <= k <= 5 - 1 <= n <= 26*k

思路

动态规划

dp[i][j]表示总长度为i，用了前j种字母可能出现的字符串的数量，可以得到如下转移方程。

dp[i][j]=\sum_{x=0}^kdp[i-x][j-1]·C_i^x

其中，x表示第j种字母出现的次数，其值可以为0,1,..k，对于每一个x，都有C_i^x种位置可以选择。
例如，x=2，表示第j种字符出现了2次，他可能出现在前i个位置中的任意两个，是一个组合问题，其结果是C_i^2。

C++代码

class Solution {
public:
    int keyboard(int k, int n) {
        vector<vector<long long>> dp(n + 1, vector<long long>(27, 0L));
        for(int i = 0;i <= 26;i++)
            dp[0][i] = 1;
        for(int i = 1;i<=n;i++) {
            for(int j = 1;j <= 26;j++) {
                for(int x = 0;x <= k;x++) {
                    if(i-x >= 0)
                    dp[i][j] += dp[i-x][j-1]*C(i,x);
                }
                dp[i][j] %=  1000000007;
            }
        }
        return dp[n][26];
    }
    long long C(int m, int n) {
        int k = 1;
        long long ans=1;
        while(k <= n) {
            ans=((m-k+1)*ans)/k;
            k++;
        }
        return ans;
    }
};