[Bzoj2431][HAOI2009]逆序对数列(前缀和优化dp)

题目链接:https://www.lydsy.com/JudgeOnline/problem.php?id=2431

dp[i][j]表示有i个数的排列方式中逆序对有j个的方案数。

因为dp这样转移为$dp[i][j]=sum_{t=j-i+1}^{j}(dp[i-1][t])$

这样i,j,t三层循环O(n^3)会T
这时发现t的范围是在j-i+1到j之间,所以如果可以处理出每次循环j时的$sum (dp[i-1][t])$就可以优化成O(n^2)。
现在思考每次dp[i][[j]时的$sum (dp[i-1][t])$,其实是dp[i][j-1]需要的;$sum (dp[i-1][t])-(j>=i-1?dp[i-1][j-i+1]:0)$。
所以可以维护一下dp[i-1][0到k]的前缀和。

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 typedef long long ll;
 4 const ll mod = 10000;
 5 ll dp[1100][1100];
 6 int main() {
 7     int n, k;
 8     scanf("%d%d", &n, &k);
 9     dp[1][0] = 1;
10     for (int i = 2; i <= n; i++) {
11         ll sum = 0;
12         for (int j = 0; j <= k; j++) {
13             sum = (sum + dp[i - 1][j]) % mod;
14             dp[i][j] = sum;
15             if (j >= i - 1)
16                 sum = (sum - dp[i - 1][j - i + 1] + mod) % mod;
17         }
18     }
19     printf("%lld
", dp[n][k]);
20 }
原文地址:https://www.cnblogs.com/sainsist/p/11130348.html