920. 播放列表的数量

题目
你的音乐播放器里有 N 首不同的歌,在旅途中,你的旅伴想要听 L 首歌(不一定不同,即,允许歌曲重复)。请你为她按如下规则创建一个播放列表:

  1. 每首歌至少播放一次。
  2. 一首歌只有在其他 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=Cn1m1+Cn1m
直观上理解为:从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(n1,l1,K)n+f(n1,l,K)(nK)
其中, 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 NL
K &gt; N K&gt;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的问题。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值