LeetCode 920 動態規劃
阿新 • • 發佈:2018-12-14
class Solution { public: int numMusicPlaylists(int N, int L, int K) { const int mod = 1e9+7; vector<vector<long long>> dp(L+1,vector<long long>(N+1,0)); dp[0][0] = 1; for(int i=1;i<=L;i++){ //當前長度 for(int j=1;j<=N;j++){ //已經使用了多少個字元 dp[i][j] = (dp[i][j]+(dp[i-1][j-1]*(N-j+1))%mod)%mod; //第一次使用,則該曲子有N-j+1種選擇 dp[i][j] = (dp[i][j]+(dp[i-1][j]*max(j-K,0))%mod)%mod; //已經使用了一次,則要滿足其他曲子的個數達到K的要求,目前已經使用的曲子是j個 dp[i][j] %= mod; } } return dp[L][N]; } };