BZOJ 2431 [HAOI2009]逆序对数列 动态规划

Description

对于一个数列{ai}，如果有i<j且ai>aj，那么我们称ai与aj为一对逆序对数。若对于任意一个由1~n自然数组成的

数列，可以很容易求出有多少个逆序对数。那么逆序对数为k的这样自然数数列到底有多少个？

Input

第一行为两个整数n，k。

Output

写入一个整数，表示符合条件的数列个数，由于这个数可能很大，你只需输出该数对10000求余数后的结果。

Sample Input

4 1

Sample Output

3

样例说明：
下列3个数列逆序对数都为1；分别是1 2 4 3 ；1 3 2 4 ；2 1 3 4；
100%的数据 n<=1000，k<=1000

HINT

传送门

用f(n,k)表示n个数刚好有k个逆序对的方案数。

假设我们已经放好了1~n-1，而且逆序对数量为k，

对于n，我们可以怎么放呢？

假设把这1~n-1个数从左往右排好了顺序，

显然，n可以放在所有n-1个数的后面，逆序对个数仍然是k；

n可以放在第n-1个数的前面，这样子逆序对个数变成了k+1；

也可以放在第n-2个数的前面，这样子逆序对个数变成了k+2；

……

放在第1个数的前面，逆序对个数变成了k+n-1。

我们可以根据上面的思路写出f(n,k)的转移式：

f(n,k)=f(n-1,k-n+1)+f(n-1,k-n+2)+……+f(n-1,k)

得到了一个O(n^3)的算法。

如何优化呢？看到了f(n,k)都从f(n-1,k)转过来，

用sum(n,k)表示f(n,k)的前k项的和，就可以直接求出每个f(n,k)

f(n,k)=sum(n-1,k)-sum(n-1,k-n)，注意k<n的情况。

并且对于每次的n，都重新建立一遍sum。

这样就可以优化到了O(n^2)。

事实上sum和f都可以滚动数组优化。

#include<bits/stdc++.h>
using namespace std;
const int 
	N=1005,
	moder=10000;
int n,K;
int f[2][N],sum[2][N];
int main(){
	cin>>n>>K;
	f[0][0]=sum[0][0]=1;
	for (int i=1;i<=K;i++) sum[0][i]=1;
	for (int i=1;i<=n;i++){
		int now=i&1,pre=now^1,tmp;
		sum[now][0]=f[now][0]=1;
		for (int j=1;j<=K;j++){
			if (j-i<0) tmp=0; else tmp=sum[pre][j-i];
			f[now][j]=(sum[pre][j]-tmp+moder)%moder;
			sum[now][j]=(sum[now][j-1]+f[now][j])%moder;
		}
	}
	cout<<f[n&1][K]<<endl;
	return 0;
}