小扣在秋日市集购买了一个古董键盘。由于古董键盘年久失修,键盘上只有 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"
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/Uh984O
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
组合数:
C(m, n) = C(m-1, n-1) + C(m-1, n)
动态规划:
dp[i][j] 表示总共长度 i ,用了前 j 种字母 可能出现的字符串的数量
转移方程:
public class Solution1 {
public static void main(String[] args) {
Solution1 solution1=new Solution1();
int k=1, n=1;
System.out.println(solution1.keyboard(k,n));
}
public int keyboard(int k, int n){
int[][] dp = new int[n+1][27];
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;
}
}
}
int ans = dp[n][26];
return ans;
}
private int C(int m, int n){
int k = 1;
int ans = 1;
while(k<=n){
ans = ((m-k+1)*ans)/k;
k++;
}
return ans;
}
}