两道序列DP

（以下来自Luogu Voldermod 的10.21 Luogu个人赛的T1题面）：

最后的战役打响了。

哈利被宣告“死亡”，伏地魔带着他的部下准备进攻霍格沃茨。但是霍格沃茨有古老的魔法保护，他们必须先摧毁这些保护。魔法保护一共有

伏地魔每秒都会穿过一层保护，他在第

1.收集

2.收集

3.使用加倍魔法

对于上面三个选择，他每秒可以可以选择一个，并可能获得能量，对于不同的选择，获得的能量也不同：

对于1.获得

对于2.获得

对于3.这一秒总共收集的能量不变（也就是这一秒不收集新的能量），但是下一秒获得的能量翻倍。但是他不能连续使用加倍魔法，而且他最多只能使用

只有他通过了这

于是，伏地魔找到了你，而你，作为精通计算机技术的麻瓜程序员，现在需要做的就是设计一个程序帮助伏地魔计算出他可以获得的最大的魔法能量的值。

最终的决战已经展开，魔法界的历史又翻过了一页……

显然DP，容易得到状态：设f[i][j][0/1]表示到了第i层，用了j次魔法，本次用（1）还是没用（0）。

那么状态转移方程就是：

设res=max (k[PreM[i]], PreS[Map[x[i]]]); （Map是离散化用的，因为值域到了10⁹）

那么f[i][j][0]=max (f[j][0]+res, f[j][1]+res*2), f[i-1][j][1]=f[i-1][j-1][0];

特别的 f[i][0][0]=f[i-1][0][0]+res。

但是注意到内存只有256MB，而f数组要开long long，算一下有300多MB。空间炸了？？

这个时候，注意到第i层只能有第i-1层转移过来，所以滚掉一维，倒序枚举就OK了。

#include <bits/stdc++.h>
using namespace std;
inline int gi () {
    int x=0,w=0; char ch=0;
    while (!(ch>='0' && ch<='9')) {if (ch=='-') w=1; ch=getchar ();}
    while (ch>='0' && ch<='9') x= (x<<3)+(x<<1)+(ch^48), ch=getchar ();
    return w?-x:x;
}

#define LL long long
#define RG register

const int M=510;
const int N=5e4+10;

map <int,int> Map;
LL Ans,f[M][2];
int n,m,tot,ty[N],b[N],x[N],k[N],PreM[N],PreS[N];

int main ()
{
    RG int i,j,res;
    n=gi (), m=gi ();
    for (i=1;i<=n;++i) {
        ty[i]=gi (), k[i]=gi ();
        if (Map.find (ty[i])==Map.end ()) Map[ty[i]]=++tot;
    }
    for (i=1;i<=n;++i) x[i]=gi ();
    for (i=1;i<=n;++i) 
        if (k[i]>k[PreM[i-1]]) PreM[i]=i;
        else PreM[i]=PreM[i-1];
    //for (i=1;i<=n;++i) cout << ty[i] << ' ';
    //cout << '\n';
    //for (i=1;i<=n;++i) cout << PreM[i] << ' ';
    // cout << '\n';1
    f[0][0]=k[1], PreS[Map[ty[1]]]=k[1], Ans=k[1];
    for (i=2;i<=n;++i) {
        PreS[Map[ty[i]]]+=k[i];
        // cout << "PreSum:" << PreS[i] << '\n';
        res=max (k[PreM[i]], PreS[Map[x[i]]]);
        // cout << "res:" << res << '\n';
        for (j=min (m,i);j>=1;--j) {
            //    cout << "j=" << j << '\n';
            f[j][0]=max (f[j][0]+res, f[j][1]+res*2);
            //cout << f[j][0] << '\n';
            f[j][1]=f[j-1][0];
            //cout << f[j][1] << '\n';
            //cout << "now:" << Ans << '\n';
            Ans=max (Ans, max (f[j][0], f[j][1]));
            //cout << "now:" << Ans << '\n';
        }
        f[0][0]+=res;
        //cout << f[0][0] << '\n';
        Ans=max (Ans, f[0][0]);
    }
    printf ("%lld\n", Ans);
    return 0;
}

考场上还Debug了好一阵。哎。。

猜你喜欢