K个逆序对数组
给出两个整数 n 和 k,找出所有包含从 1 到 n 的数字,且恰好拥有 k 个逆序对的不同的数组的个数。
逆序对的定义如下:对于数组的第i个和第 j个元素,如果满i < j且 a[i] > a[j],则其为一个逆序对;否则不是。
由于答案可能很大,只需要返回 答案 mod 109 + 7 的值。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/k-inverse-pairs-array
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
public class Solution {
private static final int MOD = 1000000007;
public int kInversePairs(int n, int k) {
int[][] dp = new int[n + 1][k + 1];
for (int i = 1; i <= n; ++i) {
dp[i][0] = 1;
}
/**
* dp[i][j]: dp[i-1][j] + dp[i-1][j-1] + ... + dp[i-1][j-i+1]
*
* dp[i][j-1] = dp[i-1][j-1] + ... + dp[i-1][j-i]
*/
for (int i = 2; i <= n; i++) {
for (int j = 1; j <= Math.min(k, i * (i - 1) / 2); j++) {
int val = (dp[i - 1][j] + MOD - ((j - i) >= 0 ? dp[i - 1][j - i] : 0)) % MOD;
dp[i][j] = (dp[i][j - 1] + val) % MOD;
}
}
return dp[n][k];
}
}
心之所向,素履以往 生如逆旅,一苇以航