1. 程式人生 > >LeetCode 920 動態規劃

LeetCode 920 動態規劃

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];
    }
};