2021.05.03播放列表的数量

最新推荐文章于 2024-10-01 23:05:47 发布

范谦之

最新推荐文章于 2024-10-01 23:05:47 发布

阅读量250

点赞数

文章标签：算法蓝桥杯 java

本文链接：https://blog.csdn.net/Zack_zc_zc/article/details/123624599

版权

2021.05.03播放列表的数量

题目描述

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

每首歌至少播放一次。
一首歌只有在其他 K 首歌播放完之后才能再次播放。

返回可以满足要求的播放列表的数量。由于答案可能非常大，请返回它模 10^9 + 7 的结果。

数据规模和约定

0 <= K < N <= L <= 100

思路

本次动态规划和之前不同的是动态规划大环境的设定不同。由题目中给出的N，L，Ｋ设定动态规划的大环境，而其中的各个状态是在该环境下的不同情况。
如在给定的ＮＬＫ情况下，已经排列了ｉ首歌曲，其中有ｊ首歌曲是不同的。并且目前排列好的歌曲满足Ｋ的要求。－－ｄｐ［ｉ］［ｊ］
状态转移方程：如代码所示：

代码

	public int numMusicPlaylists(int N, int L, int K) {
		int MOD = 1_000_000_007;
		long[][] dp = new long[L+1][N+1];
		dp[0][0] = 1; //初始化条件
		for(int i = 1; i <= L; i++) {
			for(int j = 1; j <= i && j <= N; j++) {
				//如果新安排的第i首歌曲是之前没有的，那么找dp[i-1][j-1]，共可以安排N-(j-1)种歌曲。
				dp[i][j] += dp[i-1][j-1]*(N-(j-1)); 
				//如果新安排的第i首歌曲是之前安排过的，那么找dp[i-1][j]，共可以安排j-k种（因为在第i首之前的k首不能安排，防止重复）
				dp[i][j] += dp[i-1 ][j]*Math.max(j-K, 0);
				//防止数值过大
				dp[i][j] %= MOD;
			}
		}
		return (int)dp[L][N];
    }