​LeetCode-每日一题 629. K个逆序对数组 [Java实现]

给出两个整数 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 个逆序对。

---

方法一：动态规划

从 nn 和 kk 数据范围均为 10^3 可以看出这是一道二维的动态规划题。

定义 f[i][j] 为考虑使用数值 [1, i ]，凑成逆序对数量恰好为 j 的数组个数。

不失一般性的考虑 f[i][j] 该如何计算，对第 i 个数（即数值为 i 的数）所在位置进行讨论，共有 i 种选择。

假设第 i 个数所在位置为 k，由于数值 i 为整个数组的最大值，因此数值 i 与前面所有数均不形成逆序对，与后面的所有数均形成逆序对。因此与数值 i 直接相关的逆向对的数量为 (i−1)−k，由此也得出与 i 不相关的逆序对数量为 j−(i−1−k)，而与 ii 不相关的逆序对数量由 f[i−1][x] 可得出。

一些细节：为处理负数问题，我们可以在取模之前先加一次 mod；另外需要对 j 和 i 的大小进行分情况讨论，防止数值 i 放置的位置“过于靠前”导致组成逆序对的数量超过 j。

    private static final int MOD = 1000000007;

    public int kInversePairs(int n, int k) {
        // f[i][j] 考虑使用数值 [1,i]，凑成逆序对数量恰好为 j 的数组个数。
        int[][] f = new int[n + 1][k + 1];
        int[][] sum = new int[n + 1][k + 1];
        f[1][0] = 1;
        Arrays.fill(sum[1], 1);
        for (int i = 2; i <= n; ++ i) {
            for (int j = 0; j <= k; ++ j) {
                f[i][j] = j < i ? sum[i - 1][j] : (sum[i - 1][j] - sum[i - 1][j - (i - 1) - 1] + MOD) % MOD;
                sum[i][j] = j == 0 ? f[i][j] : (sum[i][j - 1] + f[i][j]) % MOD;
            }
        }
        return f[n][k];
    }

题解来自于此