涨知识啦!
以前写算法题,都是解决各种超时TLE问题,今天遇到一个 dp 题目,非常激动——我写出了他的状态转移方程,本以为自己变六了,提交MLE之后才知道出题人考察的不止是dp, 还有空间压缩!
链接:https://ac.nowcoder.com/acm/contest/1088/K
来源:牛客网假日团队赛15
题目描述
在河上有一座独木桥,一只青蛙想沿着独木桥从河的一侧跳到另一侧。在桥上有一些石子,青蛙很讨厌踩在这些石子上。由于桥的长度和青蛙一次跳过的距离都是正整数,我们可以把独木桥上青蛙可能到达的点看成数轴上的一串整点:0,1,……,L(其中L是桥的长度)。坐标为0的点表示桥的起点,坐标为L的点表示桥的终点。青蛙从桥的起点开始,不停的向终点方向跳跃。一次跳跃的距离是S到T之间的任意正整数(包括S,T)。当青蛙跳到或跳过坐标为L的点时,就算青蛙已经跳出了独木桥。
题目给出独木桥的长度L,青蛙跳跃的距离范围S,T,桥上石子的位置。你的任务是确定青蛙要想过河,最少需要踩到的石子数。
输入描述:
第一行有一个正整数L(1<=L<=10^9),表示独木桥的长度。
第二行有三个正整数S,T,M,分别表示青蛙一次跳跃的最小距离,最大距离,及桥上石子的个数,其中1<=S<=T<=10,1<=M<=100。
第三行有M个不同的正整数分别表示这M个石子在数轴上的位置(数据保证桥的起点和终点处没有石子)。
所有相邻的整数之间用一个空格隔开。
输出描述:
只包括一个整数,表示青蛙过河最少需要踩到的石子数。
输入复制 输出复制
10 2
2 3 5
2 3 5 6 7
解题方法
这道题很明显是到动态规划,它的状态转移方程也很好写出来。如果我们把 i 位置上有石头记录在 flag 数组里,dp[i]={min(dp[i-j]+flag[i]) | s<=j<=t};
可是我提交一次之后竟然出现了大写的 MLE超内存! wdnmd什么玩意。没办法啊出错就改吧——发现这个L的范围太大了,不报错才怪。。。
在网上看到有人提到空间压缩的一个方法——2520压缩方式。
我们可以观察到1、2、3、4、5、6、7、8、9、10这十个数的最小公倍数是2520。这也就意味着小青蛙一次跳跃的距离无论是多少都可以经过2520,这样一来我们就可以将相距 2520 的两个石子 向前/向后 移动2520步。
这个路径压缩方式参透之后,配合上那个MLE的超内存算法,就可以顺利实现!
源码
#include<iostream>
#include<algorithm>
#include<cstdio>
using namespace std;
const int maxn=400000;
int a[105],d[105],flag[maxn];
int dp[maxn], l, s, t, m;
int main() {
cin>>l>>s>>t>>m;
for(int i=1; i<=m; i++)
cin>>a[i];
sort(a+1, a+m+1);
//路径压缩
for(int i=1; i<=m; i++)
d[i]=(a[i]-a[i-1])%2520; //石子距离按压缩后存入
for(int i=1; i<=m; i++) {
a[i]=a[i-1]+d[i];
flag[a[i]]=1; //记录位置
}
l=a[m]; //压缩后最后一子的位置
//初始化
//小青蛙要上岸 l+t就是压缩后的终点
for(int i=0; i<=l+t; i++)
dp[i]=m;
dp[0]=0;
//dp状态转移原型
for(int i=1; i<=l+t; i++) {
for(int j=s; j<=t; j++) {
if(i-j>=0) dp[i]=min(dp[i], dp[i-j]);
dp[i]+=flag[i];
}
}
//因为 l+t才是终点因此 l到 l+t这一段才是最后一步
int ans=m;
for(int i=l; i<l+t; i++)
ans=min(ans, dp[i]);
cout<<ans;
return 0;
}
瞟了几眼大佬的题解
- 72压缩方式——依据的是1~10这十个数最大不能表示的数的71,这种方法我也没有研究过。
- 另外一种也可以先将数据 离散化处理,再解决问题。
- 第三种就是与本文一样 通俗易懂的2520。
可对我来说仿佛接触到了内存压缩的新大陆,以前都是解决各种超时TLE问题,题目里提到的内存限制几乎没有在意过,看来有些问题或早或晚都要去处理。