你的音乐播放器里有 N 首不同的歌,在旅途中,你的旅伴想要听 L 首歌(不一定不同,即,允许歌曲重复)。请你为她按如下规则创建一个播放列表:
每首歌至少播放一次。
一首歌只有在其他 K 首歌播放完之后才能再次播放。
返回可以满足要求的播放列表的数量。由于答案可能非常大,请返回它模 10^9 + 7 的结果。
示例 1:
输入:N = 3, L = 3, K = 1
输出:6
解释:有 6 种可能的播放列表。[1, 2, 3],[1, 3, 2],[2, 1, 3],[2, 3, 1],[3, 1, 2],[3, 2, 1].
示例 2:
输入:N = 2, L = 3, K = 0
输出:6
解释:有 6 种可能的播放列表。[1, 1, 2],[1, 2, 1],[2, 1, 1],[2, 2, 1],[2, 1, 2],[1, 2, 2]
示例 3:
输入:N = 2, L = 3, K = 1
输出:2
解释:有 2 种可能的播放列表。[1, 2, 1],[2, 1, 2]
提示:
0 <= K < N <= L <= 100
思路:定义dp[i][j]表示前i首歌中有j首不同的歌的方案数,我们考虑当前第i首歌:
1)若该歌为新歌,也就是在前i-1首歌中没有出现过,则其方案数为dp[i-1][j-1]*(N-(j-1))
2)若该歌在前i-1首中出现过,则其方案数为dp[i-1][j]*(j-K)
做题是只想到了这个方法,其实和官方题解的方法一一样,官方题解中还有两种方法,感兴趣的读者可以研究一下。
class Solution {
public int numMusicPlaylists(int N, int L, int K) {
int mod=1000000007;
long[][] dp=new long[L+1][N+1];
dp[0][0]=1;
for(int i=1;i<=L;i++)
for(int j=1;j<=N;j++)
dp[i][j]=(dp[i-1][j-1]*(N-j+1)+dp[i-1][j]*Math.max(j-K, 0))%mod;
return (int)dp[L][N];
}
}