[LeetCode] （Hard） 920. Number of Music Playlists 音乐播放列表的个数

题目描述

你的音乐播放器里有 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

来源：力扣（LeetCode）

链接：https://leetcode-cn.com/problems/number-of-music-playlists

思路

这道题说是一个音乐播放器有N首歌，有个她想听L首歌（可以有重复），但需要满足两个条件，一个是每首歌都必须至少播放1次，第二个是两首重复歌的中间至少要有K首其他的歌。提示了结果可能非常巨大，需要对一个超大数取余。对于这类结果超大的数，基本不用怀疑，基本都是用动态规划 Dynamic Programming 来做，这里主要参考了 (大神 optimisea 的帖子)[https://leetcode.com/problems/number-of-music-playlists/discuss/180338/DP-solution-that-is-Easy-to-understand]。首先就是要确定 dp 的定义式，显然这里一维的 dp 数组是罩不住的，因为貌似有三个参数，N，L 和 K。但是否意味着需要个三维数组呢，其实也不用，并不关心所有的K值，但是对于N和L是必须要关注的，这里用一个二维 dp 数组，其中 dp[i][j] 表示总共放了i首歌，其中j首是不同的。下面来考虑状态转移方程，在加入一首歌的时候，此时有两种情况：

• 当加入的是一首新歌，则表示之前的 L-1 首歌中有 j-1 首不同的歌曲，其所有的组合情况都可以加上这首新歌，那么当前其实有 N-(j-1) 首新歌可以选。

• 当加入的是一首重复的歌，则表示之前的 L-1 首歌中已经有了 j 首不同的歌，那么若没有K的限制，则当前有 j 首重复的歌可以选。但是现在有了K的限制，意思是两首重复歌中间必须要有K首其他的歌，则当前只有 j-K 首可以选。而当 j<k 时，其实这种情况是为0的。<="" li="" style="box-sizing: border-box;">

综上所述可以得到状态转移方程：

   
   
 dp[i-1][j-1]*(N-(j-1)) + dp[i-1][j]*(j-k) (j > K)
 /
dp[i][j] = 
 \
 dp[i-1][j-1] x (N-(j-1)) (j <= K)

代码（C++）

   
   
class Solution {
public:
 int numMusicPlaylists(int N, int L, int K) {
 int M = 1e9 + 7;
 vector<vector<long>> dp(L + 1, vector<long>(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))) % M;
 if (j > K) {
 dp[i][j] = (dp[i][j] + dp[i - 1][j] * (j - K) % M) % M;
 }
 }
 }
 return dp[L][N];
 }
};

Github 同步地址:

https://github.com/grandyang/leetcode/issues/920

参考资料：

https://leetcode.com/problems/number-of-music-playlists/

https://leetcode.com/problems/number-of-music-playlists/discuss/178415/C%2B%2BJavaPython-DP-Solution

https://leetcode.com/problems/number-of-music-playlists/discuss/180338/DP-solution-that-is-Easy-to-understand

LeetCode All in One 题目讲解汇总(持续更新中...)

推荐阅读
零基础的前端开发初学者应如何系统地学习？

我是如何刷LeetCode的？

每日一荐 2019-10 汇总

LeetCode「简单」题目合集 （js/python/c++/java解法）