在我们编写dp方程尤其是在区间dp的方程之中,我们经常能够使用到前缀和其进行优化,主要的原因是在于在进行区间dp的时候,有时候是对前面一层状态的一个累加,在这个时候我们完全可以使用前缀和对它进行优化,不过要注意的就是这个区间的边界问题
下面我们来看一道板子题,出门左转
对于这个板子题的意思我们很容易了解,就是在找到逆序对数目为k的排列的数目.
对于这个排列,我们可以得到这样一个方程 f [ i ] [ j ] = ∑ k = j − i + 1 j f [ i − 1 ] [ k ] f[i][j] = \sum_{k=j-i+1}^{j}f[i-1][k] f[i][j]=∑k=j−i+1jf[i−1][k], f [ i ] [ j ] f[i][j] f[i][j] 表示的是在前 i i i 个数字中间逆序对数目为 k k k 的排列个数,对于这样一个关系,对于 f [ i ] [ j ] f[i][j] f[i][j] ,我们不必每次都从前一层的状态去给他从头加到尾一次,可以使用前缀和对其进行优化
(sum += f[i - 1][j]) %= p;
f[i][j] = sum;
if(j >= i - 1) sum = (sum - f[i - 1][j - i + 1] + p)%p;
要减去 f [ i − 1 ] [ j − i + 1 ] f[i-1][j-i+1] f[i−1][j−i+1]的原因是此时左边的边界已经在0点之后了,所以需要去减去这个边界的值
#include <cstdio>
#include <iostream>
#define N 1010
#define p 10000
using namespace std;
int n, k, f[N][N];
int main()
{
scanf("%d%d", &n, &k);
f[1][0] = 1;//初始条件,1的逆序为0,且只有1个排列
for (int i = 2; i <= n; i++)
{
int sum = 0;
for (int j = 0; j <= k; j++)
{
(sum += f[i - 1][j]) %= p;
f[i][j] = sum;
if(j >= i - 1) sum = (sum - f[i - 1][j - i + 1] + p)%p;
}
}
printf("%d\n", f[n][k]);
return 0;
}
希望这个可以对大家有所启发吧.