文章目录
- 1. 题目
- 2. 动态规划
- 3. 优化的DP
1. 题目
给出两个整数 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]。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/k-inverse-pairs-array
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
2. 动态规划
- f(i,j)f(i,j)f(i,j)表示
i
个数,j
个逆序对的组合方式种类 i
个数的j
个逆序对的组合,可以在i-1
个数的基础上得到- 把第
i
个数插入到前面i-1
的i
个空位上,相应的逆序数会变化 f(i - 1, j)
插在最后,逆序增加0,插在倒数第1个前面,逆序数为1,为保持总逆序数j
,那i-1
个数的逆序数为j-1
,依次类推- f(i,j)=f(i−1,j)+f(i−1,j−1)+...+f(i−1,j−i+1)f(i, j) = f(i - 1, j) + f(i - 1, j - 1) + ... + f(i - 1, j - i + 1)f(i,j)=f(i−1,j)+f(i−1,j−1)+...+f(i−1,j−i+1)
class Solution {
public:int kInversePairs(int n, int k) {//f(i,j)表示i个数,j个逆序对的组合方式//f(i, j) = f(i - 1, j) + f(i - 1, j - 1) + ... + f(i - 1, j - i + 1)int dp[n+1][k+1] = {0};int i, j, time, idx;for(i = 1; i <= n; i++)dp[i][0] = 1;for(i = 2; i <= n; i++){for(j = 1; j <= k; j++){time = i;idx = j;while(time-- && idx >= 0){dp[i][j] += dp[i-1][idx];dp[i][j] %= 1000000007;idx--;}}}return dp[n][k];}
};
时间复杂度 O(k∗n2)O(k*n^2)O(k∗n2)
3. 优化的DP
- f(i,j)=f(i−1,j)+f(i−1,j−1)+...+f(i−1,j−i+1)f(i, j) = f(i - 1, j) + f(i - 1, j - 1) + ... + f(i - 1, j - i + 1)f(i,j)=f(i−1,j)+f(i−1,j−1)+...+f(i−1,j−i+1)
- 代入
j=j-1
到上式 - f(i,j−1)=f(i−1,j−1)+f(i−1,j−2)+...+f(i−1,j−i)f(i, j-1) = f(i - 1, j-1) + f(i - 1, j - 2) + ... + f(i - 1, j - i)f(i,j−1)=f(i−1,j−1)+f(i−1,j−2)+...+f(i−1,j−i)
- 做差,f(i,j)−f(i,j−1)=f(i−1,j)−f(i−1,j−i)f(i, j)-f(i, j - 1) = f(i - 1, j) - f(i - 1, j - i)f(i,j)−f(i,j−1)=f(i−1,j)−f(i−1,j−i)
- 即 f(i,j)=f(i,j−1)+f(i−1,j)−f(i−1,j−i)f(i, j)=f(i, j - 1) +f(i - 1, j) - f(i - 1, j - i)f(i,j)=f(i,j−1)+f(i−1,j)−f(i−1,j−i)
class Solution {
public:int kInversePairs(int n, int k) {int dp[n+1][k+1] = {0};int i, j, maxk, M = 1000000007;for(i = 1; i <= n; i++)dp[i][0] = 1;for(i = 2; i <= n; i++){maxk = i*(i-1)/2;//最大的逆序数for(j = 1; j <= k && j <= maxk; j++){dp[i][j] = dp[i][j-1]%M + (M + dp[i-1][j]-(j>=i ? dp[i-1][j-i]:0))%M;//dp[i-1][j] 不一定比 dp[i-1][j-i] 大(正态分布),+M防止负数dp[i][j] %= M;}}return dp[n][k];}
};
时间复杂度 O(k∗n)O(k*n)O(k∗n)