BZOJ2431: [HAOI2009]逆序对数列 DP

最新推荐文章于 2019-05-17 20:13:00 发布

原创最新推荐文章于 2019-05-17 20:13:00 发布 · 488 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#BZOJ #DP #递推

收录于

本文介绍了一种高效的算法来计算由1到i组成的排列中产生j个逆序对的方案数，通过维护前缀和将时间复杂度优化至O(n*k)。使用C++实现并提供了完整的代码。

f[i][j]表示1~i组成的排列中，产生j个逆序对的方案数。

当新插入元素i时，可以新构成0~i-1个逆序对。

故f[i][j]=f[i-1][j-0]+f[i-1][j-1]+...+f[i-1][j-i+1]，时间复杂度o(n^2k)

用维护前缀和优化一下，时间复杂度为o(n*k)

#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int MOD=10000;
const int N=1005;
int n,k,f[N][N],g[N][N];
int main()
{
	scanf("%d%d",&n,&k);
	f[0][0]=1;
	for(int i=0;i<=k;i++) g[0][i]=1;
	for(int i=1;i<=n;i++)
	for(int j=0;j<=k;j++) 
	{
		if(j>=i) f[i][j]=(g[i-1][j]-g[i-1][j-i]+MOD)%MOD;
		else f[i][j]=g[i-1][j];
		if(j) g[i][j]=g[i][j-1]+f[i][j];
		else g[i][j]=f[i][j];
		g[i][j]%=MOD;
	}
	printf("%d\n",f[n][k]);
}