[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\n", dp[n][k]);
20 }

 

posted @ 2019-07-04 09:16  祈梦生  阅读(135)  评论(0编辑  收藏  举报