小蓝的跳跃
题目:
输入样例:
3
4 1
1 1 1 1
3 1
1 -1 1
2 1
-1 1
输出样例:
No
Yes
Yes
思路:
借鉴蓝桥云课上别人的思路:对于这个问题,我们只需要求出按照规则进行跳跃的最大值与最小值即可,如果x
在求出的最大值与最小值之间,就说明存在一个跳跃的方式能够满足最后的结果为x
。
算法思路:我们可以设定两个数组,分别为:
fmax
和fmin
,利用最简单的动态规划思想进行求解即可
- 定义边界:
fmax[0] = fmin[0] = 0
,因为题目说明小蓝的初始值为0,fmax[1] = fmin[1] = N[1]
,定义一个边界,以便后面的递推公式使用。- 递推公式:
fmax[i] = max(fmax[i - 1], fmax[i - 2]) + N[i]
, 同理:fmin[i] = min(fmin[i - 1], fmin[i - 2]) + N[i]
,利用此递推公式即可求出最大值与最小值了。- 最后进行判断的时候需要对
fmax[n + 1]
与fmin[n + 1]
进行判断。解释:如果按照常规的方式,只需要判断到第n
个位置即可,因为我们需要知道的是最大值与最小值,因此需要对最后两个位置再次进行一次筛选,即判断fmax[n]与fmax[n - 1]
这两个位置的最大值,所以我们额外进行了一次判断,并且要求数组N
的第n + 1
个位置的值为0。同理数组fmin
也是如此。
代码:
// 小蓝的跳跃
#include<bits/stdc++.h>
using namespace std;
void solve(){
int n, x;
cin>>n>>x;
vector<int> N(n + 2 , 0), fmin(n + 2, 0), fmax(n + 2, 0);
for(int i = 1;i <= n;i ++) cin>>N[i];
fmin[1] = fmax[1] = N[1];
for(int i = 2;i <= n + 1;i ++){ // 进行循环判断到 n + 1个位置,额外进行判断了一次,以便筛选出最大/最小值
fmin[i] = min(fmin[i - 1], fmin[i - 2]) + N[i];
fmax[i] = max(fmax[i - 1], fmax[i - 2]) + N[i];
}
cout<<(fmin[n + 1] <= x && fmax[n + 1] >= x ? "Yes" : "No")<<endl;
return ;
}
int main(){
int t = 1;
cin>>t;
while(t--){
solve();
}
return 0;
}
扩展练习:746. 使用最小花费爬楼梯
题目:
代码:
class Solution {
public:
int minCostClimbingStairs(vector<int>& cost) {
int costlen = cost.size();
vector<int> dp(costlen + 1, 0);
dp[0] = cost[0];
dp[1] = cost[1];
for(int i = 2;i < costlen; i++){
dp[i] = min(dp[i - 1], dp[i - 2]) + cost[i];
}
dp[costlen] = min(dp[costlen - 1], dp[costlen - 2]);
return dp[costlen];
}
};