leetcode刷题-----k个逆序对数组

难度困难132收藏分享切换为英文接收动态反馈

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

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

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

class Solution {
    public int kInversePairs(int n, int k) {
 long[][] dp = new long[n + 1][k + 1];
    if(k > n*(n - 1) / 2 || k < 0)
        return 0;
    if(k == 0 || k == n *(n - 1) / 2)
        return 1;

    int mod = 1000000007;
    dp[2][0] = 1;
    dp[2][1] = 1;
    for(int i = 3 ; i <= n ; i ++){
        dp[i][0] = 1;
        for(int j = 1 ; j <= Math.min(k, n * (n - 1) / 2); j ++){
            dp[i][j] = dp[i - 1][j] + dp[i][j - 1];
            if(j >= i)
                dp[i][j] -= dp[i - 1][j - i];
            dp[i][j] = (dp[i][j] + mod) % mod; //处理dp[i][j]为负数的情况
        }
    }
    return (int)dp[n][k];
  
    }
}