Leetcode k个逆序数组对

最新推荐文章于 2022-04-28 10:00:21 发布

18岁程序猿想当算法工程师

最新推荐文章于 2022-04-28 10:00:21 发布

阅读量210

点赞数

分类专栏：笔记

本文链接：https://blog.csdn.net/sinat_32775443/article/details/121276901

版权

笔记专栏收录该内容

3 篇文章 0 订阅

订阅专栏

class Solution {
    public int kInversePairs(int n, int k) {
        final int MOD = 1000000007;
        int[][] f = new int[2][k + 1];
        f[0][0] = 1;
        for (int i = 1; i <= n; ++i) {
            for (int j = 0; j <= k; ++j) {
                int cur = i & 1, prev = cur ^ 1;
                f[cur][j] = (j - 1 >= 0 ? f[cur][j - 1] : 0) - (j - i >= 0 ? f[prev][j - i] : 0) + f[prev][j];
                if (f[cur][j] >= MOD) {
                    f[cur][j] -= MOD;
                } else if (f[cur][j] < 0) {
                    f[cur][j] += MOD;
                }
            }
        }
        return f[n & 1][k];
    }
}