https://nanti.jisuanke.com/t/28866
题意:
每次给你n、k,问你1-n的排列恰好有k个逆序对的排列有多少种。
n、k<=5000。
POINT:
可知:dp[n][k]的递推公式为
比如dp[4][4]=dp[3][1]+dp[3][2]+dp[3][3]+dp[3][4]。
为什么dp[4][4]不能从dp[3][0]得到,因为把4放在【1-3的排列】的最前面,最多只能得到3个贡献。
所以每次dp[i][j]都是从前一个的j-i+1 到j。 当然,j-i+1要和0做比较。比如dp[4][2]=dp[3][0]+dp[3][1]+dp[3][2]。别越界了。
所以每次加的是一个区间,可以每次加一个,减一个得到。
之后5000×5000的数组开不下,由递推式可知dp只和上一个有关,所有可以用滚动数组。
因而离线。
#include <iostream>
#include <string.h>
#include <stdio.h>
#include <algorithm>
#include <queue>
#include <vector>
using namespace std;
const int maxn = 5050;
int dp[10][5010];
const int mod = 1e9+7;
struct node
{
int n,k;
int id;
bool friend operator < (node a,node b)
{
return a.n<b.n;
}
}query[maxn];
int ans[maxn];
int main()
{
int cnt=0,n,k;
while(~scanf("%d%d",&n,&k)){
query[++cnt].n=n;
query[cnt].k=k;
query[cnt].id=cnt;
if(n==0) break;
}
sort(query+1,query+1+cnt);
dp[0][0]=1;
int p=1;
int now=1;
for(int i=1;i<=5000;i++){
int sum=0;
for(int j=0;j<=5000;j++){
sum=(sum+dp[p^1][j])%mod;
if(j-i>=0)
sum=(sum-dp[p^1][j-i]+mod)%mod;
dp[p][j]=sum;
}
while(query[now].n==i){
ans[query[now].id]=dp[p][query[now].k];
now++;
}
p=p^1;
}
for(int i=1;i<=cnt;i++){
printf("%d\n",ans[i]);
}
return 0;
}