leetcode629.K个逆序对数组

给出两个整数 n 和 k，找出所有包含从 1 到 n 的数字，且恰好拥有 k 个逆序对的不同的数组的个数。

逆序对的定义如下：对于数组的第i个和第 j个元素，如果满i < j且 a[i] > a[j]，则其为一个逆序对；否则不是。

由于答案可能很大，只需要返回 答案 mod 109 + 7 的值。

示例 1:

输入: n = 3, k = 0
输出: 1
解释: 
只有数组 [1,2,3] 包含了从1到3的整数并且正好拥有 0 个逆序对。
示例 2:

输入: n = 3, k = 1
输出: 2
解释: 
数组 [1,3,2] 和 [2,1,3] 都有 1 个逆序对。
说明:

n 的范围是 [1, 1000] 并且 k 的范围是 [0, 1000]。

思路：二维动态规划

dp[3][3] = 1 ====> [3, 2, 1]，现在求dp[4][3],也就是在原有3个数的任意排列下，插入4这个数字，所以4可以插入4个位置，[x x x 4],[x x 4 x],[x 4 x x],[4 x x x]，

1. [x x x 4] 如果将4插入最末尾，则和原来的[1 2 3]只需要产生3个逆序对，4的插入不产生新的逆序对,dp[3][3]
2. [x x 4 x] 4的插入将会产生1个逆序对，所以[1 2 3]只需要产生2个逆序对, dp[3][2]
3. [x 4 x x] 4的插入将会产生2个逆序对，所以[1 2 3]只需要产生1个逆序对, dp[3][1]
4. [4 x x x] 4的插入将会产生3个逆序对，所以[1 2 3]只需要产生0个逆序对, dp[3][0]
   dp[4][3] = dp[3][3]+dp[3][2]+dp[3][1]+dp[3][0]

d p [ i ] [ j ] = ∑ k = 0 m i n { j , i − 1 } d p [ i − 1 ] [ j − k ] dp[i][j] = \sum_{k=0}^{min\{j, i-1\}}dp[i-1][j-k] dp[i][j]=k=0∑min{j,i−1}​dp[i−1][j−k]
展开得：
$$\begin{gathered} dp[i][j]=dp[i-1][j]+dp[i-1][j-1]+...+dp[i-1][j-i+1]s.t.(j>i-1) \\ dp[i][j-1]=dp[i-1][j-1]+dp[i-1][j-2]+...+dp[i-1][j-i]s.t.(j>i-1) \end{gathered}$$
两式相减得：
$$dp[i][i]-dp[i][j-1]=dp[i-1][j]-dp[i-1][j-i]$$
即：$$dp[i][i]=dp[i-1][j]+dp[i][j-1]-dp[i-1][j-i]s.t.(j>i-1)$$
即：$$dp[i][j]=\{\begin{array}{ll}dp[i-1][j]+dp[i][j-1]-dp[i-1][j-i]& \text{j>i-1}\\ dp[i-1][j]+dp[i][j-1]& \text{else}\end{array}$$

class Solution {
public:
    const long long int MOD = 1e9+7;
    int kInversePairs(int n, int k) {
    	// dp[i][j] 表示1~i中产生j个逆序对的个数
        std::vector<std::vector<int>> dp(n+1, std::vector<int>(k+1, 0));
        dp[0][0] = 1;
        for(int i=1;i<=n;i++){
            dp[i][0] = 1;
            for(int j=1;j<=k;j++){
                dp[i][j] = j>i-1 ? dp[i-1][j] + dp[i][j-1] - dp[i-1][j-i] : dp[i-1][j] + dp[i][j-1];
                dp[i][j] = dp[i][j]<0?(dp[i][j]+MOD)%MOD:dp[i][j]%MOD;
            }
        }
        return (int)dp[n][k];
    }
};