用到动态规划
力扣 70. 爬楼梯
假设你正在爬楼梯。需要 n 阶你才能到达楼顶。
每次你可以爬 1 或 2 个台阶。你有多少种不同的方法可以爬到楼顶呢?
注意:给定 n 是一个正整数。
示例 1:
输入: 2
输出: 2
解释: 有两种方法可以爬到楼顶。
1. 1 阶 + 1 阶
2. 2 阶
示例 2:
输入: 3
输出: 3
解释: 有三种方法可以爬到楼顶。
1. 1 阶 + 1 阶 + 1 阶
2. 1 阶 + 2 阶
3. 2 阶 + 1 阶
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/climbing-stairs
代码:
class Solution {
public int climbStairs(int n) {
int[] dp=new int[n+2];
dp[1]=1;
dp[2]=2;
for(int i=3;i<=n;i++){
dp[i]=dp[i-1]+dp[i-2];
}
return dp[n];
}
}
分析:
类似斐波拉西数列
台阶数 方法数
1 1
2 2
3 1+2=3
4 2+3=5
5 3+5=8
注意创建dp数组时,大小为n+2,因为数组下标是从0开始的(虽然没有用),该代码中dp数组下标就是阶梯数,值就是方法数
力扣 198. 打家劫舍
代码:
class Solution {
public int rob(int[] nums) {
int N = nums.length;
if(N==0)return 0;//没有房子,返回0
int[] S=new int[N];
if(N==1){//有一个房子返回,第一个房子的价值
S[0]=nums[0];
return S[0];
}
if(N==2){//有两个房子,返回前两个房子价值最大的值
S[1]=Math.max(nums[0],nums[1]);
return S[1];
}
//N大于2,房子至少有三个
S[0]=nums[0];
S[1]=Math.max(nums[0],nums[1]);
for(int i=2;i<N;i++){
S[i]=Math.max(S[i-2]+nums[i],S[i-1]);
}
return S[N-1];
}
}
分析:
设S为收益数组,S[0]表示一个房子的收益,S[1]表示前两个房子的收益,S[2]表示前三个房子的收益,以此类推
首先把三N=0,1,2三种特殊情况考虑进来,要考虑2是因为i=2开始,然后是考虑N>2,即房子至少有三个的情况,
假设有三个房子,那么考虑偷不偷第三个房子,如果要偷,那么收益就是第三个房子的收益+S[0](前一个房子的最大收益,直接去掉第二个房子的收益,因为第二个房子和第三个相邻),如果不偷,收益就是S[1](前两个房子的最大收益),然后考虑两种方式的最大值
假设有四个房子,那么考虑偷不偷第四个房子,如果要偷,那么收益就是第四个房子的收益+S[1](前两个房子的最大收益,直接去掉第三个房子的收益,因为第三个房子和第四个相邻),如果不偷,收益就是S[2](前三个房子的最大收益),然后考虑两种方式的最大值
代码体现就是:S[i]=Math.max(S[i-2]+nums[i],S[i-1]);