P2513 [HAOI2009]逆序对数列

题目地址

题目大意

求出一个长度为$n$的全排列中有多少个逆序对为$k$的全排列。

思路

我们很容易想出一个朴素的状态:
$dp[i][j]$为长度为$i$的全排列中逆序对为$j$的数量。
那么转移方程也很好想，我们考虑把$i$插入到一个为$i-1$的全排列中，逆序对的变化，当把$i$放在最后面时，逆序数不变，放在倒数第$1$个的前面，逆序对增加了$1$，放在倒数第$2$个的前面，逆序数增加了$2$。那么总结起来就是$dp[i][j]={\sum }_{k=max(0,j-i+1)}^{j}dp[i-1][k]$

但是这个方程的复杂度是$O(k\ast n^2)$,无法通过此题。

观察上面的式子，我们发现$dp[i][j]$是由$dp[i-1]$中的某一段的和转移过来的，那么我们考虑前缀和优化，每次得出dp[i][j]后对它进行前缀和处理，然后对于dp[i+1]直接O(1)得出每一项即可。最后一次就不进行前缀和处理了。

代码：

#include<bits/stdc++.h>
using namespace std;
const int N = 1005;
int dp[N][N];// dp[i][j] 为长度为i的全排列，逆序数为j
int main() {
	int n, m;
	cin >> n >> m;
  auto add = [&](int i, int a, int b) {
    if(a <= 0) return dp[i][b];
    return ((dp[i][b] - dp[i][a-1]) % 10000 + 10000) % 10000;
  } ;
	for(int i = 1; i <= n; i++){
		dp[i][0] = 1;
		for(int j = 1; j <= m; j++) {
      dp[i][j] = add(i-1, j - i + 1, j);
		}
    if(i == n) break;
    for(int j = 1; j <= m; j++) {
      dp[i][j] = (dp[i][j-1] + dp[i][j]) % 10000;
    }
  }
	cout << dp[n][m] << endl;
}