题目
你的音乐播放器里有 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
【困难】
【分析】联想数学上的一个公式:
C
n
m
=
C
n
−
1
m
−
1
+
C
n
−
1
m
C_{n}^{m}=C_{n-1}^{m-1}+C_{n-1}^{m}
Cnm=Cn−1m−1+Cn−1m
直观上理解为:从n个中取m个=从n-1个中取m-1个(假设特定的某个取到)+ 从n-1个中取m个(假设特定的某个未取到,则需要从剩下的n-1个中再取m个)
本题的递归公式:
f
(
n
,
l
,
K
)
=
f
(
n
−
1
,
l
−
1
,
K
)
∗
n
+
f
(
n
−
1
,
l
,
K
)
∗
(
n
−
K
)
f(n,l,K)=f(n-1,l-1,K)*n+f(n-1,l,K)*(n-K)
f(n,l,K)=f(n−1,l−1,K)∗n+f(n−1,l,K)∗(n−K)
其中,
f
(
n
,
l
,
K
)
f(n,l,K)
f(n,l,K)表示n首个构成l首歌的歌单,任意两首相同的歌之间的间隔超过K,本题中K是常数。
直观理解为:(可重复抽取)当n-1个中取出l-1个时,特定的某个的取法是n个,所以乘以n;当n个中取出l-1个时,歌单中的最后一首的取出可能会取到跟前面重复的(因为前面是从n个中取的),所以要满足最后一个的取出不能与前k个的相同,所以有n-k种取法。
要注意,每首歌至少播放一次,说明
N
≤
L
N \leq L
N≤L;
当
K
>
N
K>N
K>N时输出应该为
0
0
0。
当
n
=
=
l
n==l
n==l时,输出应该为
n
!
n!
n!
所以创建循环
n
:
[
K
+
1
,
N
]
,
l
:
[
n
,
L
]
n: [K+1,N] , l:[n,L]
n:[K+1,N],l:[n,L]
class Solution:
def numMusicPlaylists(self, N: int, L: int, K: int) -> int:
#N=2;L=3;K=0
import math
#res=[[0]*(L+1)]*(N+1) #copy和deepcopy的问题要注意,否则就会bug
res=[[0]*(L+1) for _ in range(N+1)]
for n in range(K+1,N+1):
for l in range(n,L+1):
if n==l:
res[n][l]=math.factorial(n)
else:
res[n][l]=res[n-1][l-1]*n + res[n][l-1]*(n-K)
return res[N][L]%(10**9+7)
最后要注意python中的copy和deepcopy的问题。