2021.11.11 力扣-K个逆序对数组

题目描述：

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

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

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

方法一（自己的方法）：

class Solution {
public:
    //f[i][j]表示包含从 1 到 i 的数字，且恰好拥有 j 个逆序对的不同的数组的个数
    long long f[1001][1001] = { 0 };
    int kInversePairs(int n, int k) {
        if (k == 0) return 1;
        f[1][0] = 1;
        //count表示包含从 1 到 i - 1 的数字，最多能有count个逆序对的不同的数组的个数
        int count = 1;
        for (int i = 1; i <= n; i++)
        {
            int j;
            for (j = 0; j <= k; j++)
            {
                for (int m = max(0, j - i + 1); m <= min(j, count); m++)
                {
                    f[i][j] += f[i - 1][m];
                    //每次计算出来的数值都需要求余一次，避免出现溢出
                    f[i][j] %= (int)(1e9 + 7);
                }
            }
            //更新count的值
            count = count + i - 1;
        }
        int ans = f[n][k] % (int)(1e9 + 7);
        return ans;
    }
};

看到题目后第一时间想到的是dfs，但是题目中又说到答案可能数值很大，担心dfs可能会超时，因此最后还是选择使用dp来解。

在纸上演算了半天后终于发现了一点规律：

 如图，我列了n = 3和n = 4的两种情况，发现n = 4中的排列方式可以分为四种，每种排列可以看做是一个数字（首元素）和 n = 3的所有排列方式的结合。

若首元素是1，那么对应的k值和n = 3相同为0，1， 1， 2， 2， 3；若首元素是3，先将后面的3个数的排列看成是n = 3的情况，然后由于1和2排在3的后面，所以每个排列所对应的k值应该加2，所以首元素是3的这些排列所对应的k值为2， 3， 3， 4， 4， 5。

由此我推断出这应该是一个普遍适应的规律（反正做题嘛，不用费心去证明），再经过反复推导终于确定了递推关系式。