给出两个整数 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]。
思路:二维动态规划
dp[3][3] = 1 ====> [3, 2, 1],现在求dp[4][3],也就是在原有3个数的任意排列下,插入4这个数字,所以4可以插入4个位置,[x x x 4],[x x 4 x],[x 4 x x],[4 x x x],
- [x x x 4] 如果将4插入最末尾,则和原来的[1 2 3]只需要产生3个逆序对,4的插入不产生新的逆序对,dp[3][3]
- [x x 4 x] 4的插入将会产生1个逆序对,所以[1 2 3]只需要产生2个逆序对, dp[3][2]
- [x 4 x x] 4的插入将会产生2个逆序对,所以[1 2 3]只需要产生1个逆序对, dp[3][1]
- [4 x x x] 4的插入将会产生3个逆序对,所以[1 2 3]只需要产生0个逆序对, dp[3][0]
dp[4][3] = dp[3][3]+dp[3][2]+dp[3][1]+dp[3][0]
d
p
[
i
]
[
j
]
=
∑
k
=
0
m
i
n
{
j
,
i
−
1
}
d
p
[
i
−
1
]
[
j
−
k
]
dp[i][j] = \sum_{k=0}^{min\{j, i-1\}}dp[i-1][j-k]
dp[i][j]=k=0∑min{j,i−1}dp[i−1][j−k]
展开得:
d
p
[
i
]
[
j
]
=
d
p
[
i
−
1
]
[
j
]
+
d
p
[
i
−
1
]
[
j
−
1
]
+
.
.
.
+
d
p
[
i
−
1
]
[
j
−
i
+
1
]
s
.
t
.
(
j
>
i
−
1
)
d
p
[
i
]
[
j
−
1
]
=
d
p
[
i
−
1
]
[
j
−
1
]
+
d
p
[
i
−
1
]
[
j
−
2
]
+
.
.
.
+
d
p
[
i
−
1
]
[
j
−
i
]
s
.
t
.
(
j
>
i
−
1
)
dp[i][j] = dp[i-1][j] + dp[i-1][j-1] + ...+dp[i-1][j-i+1]\space\space s.t.(j>i-1) \\dp[i][j-1] = dp[i-1][j-1] + dp[i-1][j-2] + ...+dp[i-1][j-i]\space\space s.t.(j>i-1)
dp[i][j]=dp[i−1][j]+dp[i−1][j−1]+...+dp[i−1][j−i+1] s.t.(j>i−1)dp[i][j−1]=dp[i−1][j−1]+dp[i−1][j−2]+...+dp[i−1][j−i] s.t.(j>i−1)
两式相减得:
d
p
[
i
]
[
i
]
−
d
p
[
i
]
[
j
−
1
]
=
d
p
[
i
−
1
]
[
j
]
−
d
p
[
i
−
1
]
[
j
−
i
]
dp[i][i]-dp[i][j-1] = dp[i-1][j]-dp[i-1][j-i]
dp[i][i]−dp[i][j−1]=dp[i−1][j]−dp[i−1][j−i]
即:
d
p
[
i
]
[
i
]
=
d
p
[
i
−
1
]
[
j
]
+
d
p
[
i
]
[
j
−
1
]
−
d
p
[
i
−
1
]
[
j
−
i
]
s
.
t
.
(
j
>
i
−
1
)
dp[i][i]=dp[i-1][j]+dp[i][j-1]-dp[i-1][j-i] \space\space s.t.(j>i-1)
dp[i][i]=dp[i−1][j]+dp[i][j−1]−dp[i−1][j−i] s.t.(j>i−1)
即:
d
p
[
i
]
[
j
]
=
{
d
p
[
i
−
1
]
[
j
]
+
d
p
[
i
]
[
j
−
1
]
−
d
p
[
i
−
1
]
[
j
−
i
]
j>i-1
d
p
[
i
−
1
]
[
j
]
+
d
p
[
i
]
[
j
−
1
]
else
dp[i][j]= \begin{cases} dp[i-1][j]+dp[i][j-1]-dp[i-1][j-i]& \text{j>i-1}\\ dp[i-1][j]+dp[i][j-1]& \text{else} \end{cases}
dp[i][j]={dp[i−1][j]+dp[i][j−1]−dp[i−1][j−i]dp[i−1][j]+dp[i][j−1]j>i-1else
class Solution {
public:
const long long int MOD = 1e9+7;
int kInversePairs(int n, int k) {
// dp[i][j] 表示1~i中产生j个逆序对的个数
std::vector<std::vector<int>> dp(n+1, std::vector<int>(k+1, 0));
dp[0][0] = 1;
for(int i=1;i<=n;i++){
dp[i][0] = 1;
for(int j=1;j<=k;j++){
dp[i][j] = j>i-1 ? dp[i-1][j] + dp[i][j-1] - dp[i-1][j-i] : dp[i-1][j] + dp[i][j-1];
dp[i][j] = dp[i][j]<0?(dp[i][j]+MOD)%MOD:dp[i][j]%MOD;
}
}
return (int)dp[n][k];
}
};