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