LeetCode in Kotlin
920. Number of Music Playlists
Hard
Your music player contains n different songs. You want to listen to goal songs (not necessarily different) during your trip. To avoid boredom, you will create a playlist so that:
- Every song is played at least once.
- A song can only be played again only if
kother songs have been played.
Given n, goal, and k, return the number of possible playlists that you can create. Since the answer can be very large, return it modulo 109 + 7.
Example 1:
Input: n = 3, goal = 3, k = 1
Output: 6
Explanation: There are 6 possible playlists: [1, 2, 3], [1, 3, 2], [2, 1, 3], [2, 3, 1], [3, 1, 2], and [3, 2, 1].
Example 2:
Input: n = 2, goal = 3, k = 0
Output: 6
Explanation: There are 6 possible playlists: [1, 1, 2], [1, 2, 1], [2, 1, 1], [2, 2, 1], [2, 1, 2], and [1, 2, 2].
Example 3:
Input: n = 2, goal = 3, k = 1
Output: 2
Explanation: There are 2 possible playlists: [1, 2, 1] and [2, 1, 2].
Constraints:
0 <= k < n <= goal <= 100
Solution
class Solution {
fun numMusicPlaylists(n: Int, l: Int, k: Int): Int {
val dp = Array(l) { LongArray(n + 1) }
for (i in 0 until l) {
dp[i].fill(-1)
}
return helper(0, l, 0, n, k, dp).toInt()
}
private fun helper(songNumber: Int, l: Int, usedSong: Int, n: Int, k: Int, dp: Array<LongArray>): Long {
if (songNumber == l) {
return if (usedSong == n) 1 else 0
}
if (dp[songNumber][usedSong] != -1L) {
return dp[songNumber][usedSong]
}
val ans: Long = if (songNumber < k) {
(n - usedSong) * helper(songNumber + 1, l, usedSong + 1, n, k, dp)
} else if (usedSong == n) {
(usedSong - k) * helper(songNumber + 1, l, usedSong, n, k, dp)
} else {
(
(n - usedSong) * helper(songNumber + 1, l, usedSong + 1, n, k, dp) +
(usedSong - k) * helper(songNumber + 1, l, usedSong, n, k, dp)
)
}
val mod = 1e9.toInt() + 7
dp[songNumber][usedSong] = ans % mod
return ans % mod
}
}