要求最小的步数,是不是非常easy想到用dp啊?
我一開始的做法是,当找到了一个可以从它延伸到更远的位置,就把这个位置和最远位置的步数都更新一下,结果超时了。
事实上这样不仅是超时的,并且是错误的。由于这段距离里的非常多点,事实上是上一步就能到达的。
事实上应该更新的仅仅有哪些新的可以到达的点,那哪些点是新的能到达的点呢?如果这次可以延伸更远的点是i。它的前进步数是A[i],上一次可以到达的最远的点是mmax,那新的能到达的点应该是(mmax, i+A[i] ]这个之间的这些位置。时间复杂度一下子变成线性的了。
class Solution {
public:
int jump(int A[], int n) {
if(n <= 1) return 0;
vector<int> steps(n, INT_MAX);
steps[0] = 0;
int mmax=0;
for(int i=0;i<n-1;i++){
if(i+A[i]>mmax){
for(int j=mmax+1;j<=i+A[i]&&j<n;j++)
steps[j] = steps[i]+1;
if(i+A[i]>=n-1)
steps[n-1] = min(steps[n-1], steps[i]+1);
mmax = i+A[i];
}
}
return steps[n-1];
}
};