在一个排列中,如果一对数的前后位置与大小顺序相反,即前面的数大于后面的数,那么它们就称为一个逆序。一个排列中逆序的总数就称为这个排列的逆序数。
如2 4 3 1中,2 1,4 3,4 1,3 1是逆序,逆序数是4。
1-n的全排列中,逆序数最小为0(正序),最大为n*(n-1) / 2(倒序)
给出2个数n和k,求1-n的全排列中,逆序数为k的排列有多少种?
例如:n = 4 k = 3。
1 2 3 4的排列中逆序为3的共有6个,分别是:
1 4 3 2
2 3 4 1
2 4 1 3
3 1 4 2
3 2 1 4
4 1 2 3
由于逆序排列的数量非常大,因此只需计算并输出该数 Mod 10^9 + 7的结果就可以了。
Input
第1行:一个数T,表示后面用作输入测试的数的数量。(1 <= T <= 10000) 第2 - T + 1行:每行2个数n,k。中间用空格分隔。(2 <= n <= 1000, 0 <= k <= 20000)
Output
共T行,对应逆序排列的数量 Mod (10^9 + 7)
Input示例
1 4 3
Output示例
6
由于n*k比较小(20M),尝试dp。
考虑将数从小到大加入序列。那么加入第一个数,显然对逆序对贡献是0。加入第2个数,贡献可以是0或1,这取决于加入的位置。那么加入第i个数,贡献可以是0,1,2,3...,i-1。那么就得到了方程:
f i,j=∑(l<i)f i-1,j-l%mod
这个复杂度是O(n^2k)的,但显然可以用前缀和优化。
那么预处理的复杂度就是O(nk),每个询问复杂度O(1)
附代码
#include<bits/stdc++.h>
using namespace std;
const int mod=1e9+7;
int n,k,t;
int f[1010][20010],sum[20100];
int main(){
f[1][0]=1;
sum[1]=1;
for(int i=2;i<=1000;i++){
for(int j=0;j<=min(i*(i-1)/2,20000);j++){
sum[j]=f[i-1][j];
if(j) sum[j]+=sum[j-1];
sum[j]%=mod;
f[i][j]=sum[j];
if(j>=i) f[i][j]-=sum[j-i];
f[i][j]+=mod;f[i][j]%=mod;
}
}
scanf("%d",&t);
while(t--){
scanf("%d%d",&n,&k);
printf("%d\n",f[n][k]);
}
return 0;
}