题解 | #又见台阶#

又见台阶

http://www.nowcoder.com/practice/fac4dc5774204806b7f07ac9e00fb073

又见台阶

一共又n层台阶有些台阶上有积水,牛牛一开始在第0层,它每次可以跳奇数层台阶,他想跳到第n曾,但他不想在跳跃的过程中跳到有积水的台阶,现在已知有m个台阶上有积水,问牛牛在不跳到积水台阶的情况下跳到第n层有多少种跳法, 答案对取模

案例
输入:9,3,[1,3,5]
返回值:2
说明:
因为1,3,5都不能走,所以第一步可以跳到第7层或者第9层
所以一共两种方案:

  1. 第一步跳7,第二步跳1,第三步跳1
  2. 第一步跳9

方法一 模拟

遍历所有台阶对于第i个有3种情况

  • 如果台阶上有积水,则跳过当前台阶不统计当前台阶的次数
  • 如果台阶上没有积水,且当前的台阶是奇数,则当前台阶的次数最少为1因为可以直接从0跳到当前台阶,并加上前面所有偶数台阶的次数,因为可以从偶数台阶跳奇数次跳到奇数台阶
  • 如果台阶上没有积水,且当前的台阶是偶数,则与奇数次相反,把前面所有奇数的台阶次数加起来,因为跳到偶数台阶的情况只有一种就是上一次的台阶是奇数阶,奇+奇=偶
class Solution {
public:
    long long dp[100010];
    long long  mod = 1e9 + 7;
    int solve(int n, int m, vector<int>& a) {
        // write code here
        dp[0] = 1;
        int id = 0;
        for(int i = 1; i <= n; i ++){
           if(id < m && a[id] == i){ //如果遇到积水的格子则跳过
               id ++; continue;
           }
           int t = 1;
           while(i - t >= 0){ //把前面所有能踩到的格子加起来
               dp[i] += dp[i - t];
               dp[i] %= mod;
               t += 2;
           }
           dp[i] %= mod;
        }
        return dp[n] % mod;
    }
};

时间复杂度: 遍历所有的台阶i,对于i台阶会使用 次去统计答案所以总复杂度为
空间复杂度: dp数组统计每一位到的方案数,最多会记录n个台阶的方案数

方法二 迭代

主要的思想和方法一一样类似对方法一加了优化,定义两个变量一个存储奇数台阶odd的总方案数一个存储偶数台阶even的总方案数,当遍历到第i个台阶时如果当前台阶是奇数台阶则将更新为答案,并且将奇数变量加上当前的答案,偶数则相反
图片说明

class Solution {
public:
    long long mod = 1e9 + 7;
    int solve(int n, int m, vector<int>& a) {
        if(a[m - 1] == n) return 0;
        long long odd = 0, even = 0;
        int id = 0, ans;
        for(int i = 1; i <= n; i ++){
            if(id < m && a[id] == i){
                id ++; continue;
            }
            if(i % 2) { //为奇数时说明当前的方案数一定+1并且选取前面偶数为的所有方案数
                ans = 1 + even;
                odd += even + 1;
            }else{ //偶数相反 将前面奇数位加起来
                ans = odd;
                even += odd;
            }
            odd %= mod; even %= mod; ans %= mod;
        }
        return ans;
    }
};

时间复杂度: 只需要遍历一遍n即可
空间复杂度: 只使用若干个变量

全部评论

相关推荐

点赞 评论 收藏
分享
头像
11-09 17:30
门头沟学院 Java
TYUT太摆金星:我也是,好几个华为的社招找我了
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务