$Luogu P2029$ 跳舞题解

一道不是十分水的$dp$.

首先我们考虑$dp$方程的构造。起初我定义的状态是$dp_{i,j}$表示前$i$个格子，总共跳了$j$次的最大得分。但事实上它并不可以转移，因为我们不知道新的一轮操作从之间的哪个格子算起。

那么状态转移方程就出来了，我们把第一维改成本次跳到第$i$个格子上，包括本次在内总共跳了$j$次的最大得分，那么转移的时候，由于本次一定要跳到$i$上（如状态中所定义），所以不用分类讨论。方程就是：\[dp_{i,j}=\max\{dp_{k,j-1}-\rm{Sum(k + 1, i-1)}\}+A_i\]
其中$0 \leq k < i(\text{不能两次跳到同一个格子上所以右区间为开区间})$，$\rm{Sum(l,r)}\mathcal{=\sum\limits_{i=l}^{r}A_i}$

代码大概是这样$(\rm{30pts})$：

#include <cstdio>
#include <iostream>

#define MAXN 5010 

using namespace std ; int i, j, k, Ans ;
int N, T, S[MAXN], dp[MAXN][MAXN], A[MAXN], B[MAXN] ;

int main(){
    cin >> N >> T ;
    for (i = 1 ; i <= N ; ++ i)
        scanf("%d", &A[i]), S[i] = S[i - 1] + A[i] ;
    for (i = 1 ; i <= N ; ++ i) scanf("%d", &B[i]) ;
    for (i = 0 ; i <= N ; ++ i)
        for (j = 0 ; j <= N ; ++ j)
            dp[i][j] = -192608170 ; dp[0][0] = 0 ;
    for (i = 1 ; i <= N ; ++ i)
        for (j = 1 ; j <= i ; ++ j){
            for (k = 0 ; k < i ; ++ k)
                dp[i][j] = max(dp[i][j], dp[k][j - 1] - S[i - 1] + S[k] + A[i]) ;
            if (j % T == 0) dp[i][j] += B[i] ; Ans = max(Ans, dp[i][j]) ;
        }
    cout << Ans << endl ; return 0 ;
}

但是我们发现，这个复杂度是$\Theta(n^3)$的，于是选择优化。$dp$优化的老套路就是：

优化状态维数
优化转移复杂度

而此处我们不可以优化状态了，所以考虑优化转移复杂度。转移的复杂度是$\Theta(n)$的，我们考虑可否$\Theta(1)$转移，最终使得总复杂度为$\Theta(n^2) \times \Theta(1) \leq O(n^2)$

从状态转移方程入手，我们发现有关于$k$是满足单调性的。所以不妨我们记录一下每次的$k$，即把$dp[k][j-1]+ S[k]$中的最大值存储下来，从而达到$\Theta(1)$转移的目的。

此处笔者使用了比较玄学的存储方式……类似刷表……当然这个地方有多种的优化方式啦～

完整版$code$（700～800ms）:

#include <cstdio>
#include <iostream>

#define MAXN 5010 

using namespace std ; int i, j, k, p, Ans ;
int N, T, Last[MAXN], S[MAXN], dp[MAXN][MAXN], A[MAXN], B[MAXN] ;

int main(){
    cin >> N >> T ;
    for (i = 1 ; i <= N ; ++ i)
        scanf("%d", &A[i]), S[i] = S[i - 1] + A[i] ;
    for (i = 1 ; i <= N ; ++ i) scanf("%d", &B[i]) ;
    for (i = 0 ; i <= N ; ++ i)
        for (j = 0 ; j <= N ; ++ j)
            dp[i][j] = -192608170 ; dp[0][0] = 0 ;
    for (j = 1 ; j <= N ; ++ j){
        for (i = j ; i <= N ; ++ i){
            p = Last[i - j], Last[i - j] = 0 ;
            dp[i][j] = p - S[i - 1] + A[i] ;
            if (j % T == 0) dp[i][j] += B[i] ; Ans = max(Ans, dp[i][j]) ;
            Last[i - j] = max(Last[i - j - 1], dp[i][j] + S[i]) ;
        }
    }
    cout << Ans << endl ; return 0 ;
}

~~毒瘤常数优化后被艹到龟速的版本~~（1100ms +）：

#include <cstdio>
#include <cstring>
#include <iostream>

#define max Max
#define MAXN 5010
#define Inf 19260817

using namespace std ; int i, j, k, p, t, Ans ;
int N, T, Last[MAXN], S[MAXN], dp[MAXN][MAXN], A[MAXN], B[MAXN] ;

inline int Max(int a, int b){
    return a & ((b - a) >> 31) | b & ( ~ (b - a) >> 31) ;
}
inline int qr(){
    int res = 0 ; char c = getchar() ;
    while (!isdigit(c)) c = getchar() ;
    while (isdigit(c)) res = (res << 1) + (res << 3) + c - 48, c = getchar() ;
    return res ;
}
int main(){
    cin >> N >> T ;
    for (i = 0 ; i <= N ; ++ i)
        for (j = 0 ; j <= N ; ++ j)
            dp[i][j] = -Inf ; dp[0][0] = 0 ;
    for (i = 1 ; i <= N ; ++ i)
        A[i] = qr(), S[i] = S[i - 1] + A[i] ;
    for (i = 1 ; i <= N ; ++ i) B[i] = qr() ;
    for (j = 1 ; j <= N ; ++ j){
        for (i = j ; i <= N ; ++ i){
            t = i - j, p = Last[t], dp[i][j] = p - S[i - 1] + A[i] ;
            if (!(j % T)) dp[i][j] += B[i] ; Ans = max(Ans, dp[i][j]), Last[t] = max(Last[t - 1], dp[i][j] + S[i]) ;
        }
    }
    cout << Ans << endl ; return 0 ;
}

唉，先有常数后有天，反向优化$Sun$神仙啊

原文地址：https://www.cnblogs.com/pks-t/p/10159036.html

时间： 2024-10-26 04:59:40

$Luogu P2029$ 跳舞 题解

$Luogu P2029$ 跳舞题解