每日一题——LeetCode1652.拆炸弹

方法一 分类讨论

  • k=0,直接返回一个和原数组长度一样但元素都为0的数组
  • k>0,先求出前k个元素的和preSum,初始化index=k,每次循环preSum减去当前元素code[i],再加上code[index],保证preSum的值始终是当前元素后k个元素的和,如果index到了数组末尾需要跳转到数组开头
  • k<0,同理,只是index记录的是需要减去的元素的位置,preSum先减code[index],然后加上code[i]的前一个元素,preSum始终记录的是code[i]前k个元素的和

写完后发现k>0 ,k<0方法其实是一样的,考虑是否能将这两种情况合并,缩减代码

javascript 复制代码
var decrypt = function(code, k) {
    if(k===0) return new Array(code.length).fill(0)
    let res=[]
    if(k>0){
        let preSum=0,index=k
        for(let i=0;i<k;i++){
            preSum+=code[i]
        }
        for(let i=0;i<code.length;i++){
            preSum-=code[i]
            preSum+=code[index]
            if(index==code.length-1){
                index=-1
            }
            index++
            res.push(preSum)
        }
    }else{
       let index=code.length+k,preSum=0
       for(let i=index;i<code.length;i++){
           preSum+=code[i]
       } 
       res.push(preSum)
       for(let i=0;i<code.length-1;i++){
           preSum-=code[index]
           if(index==code.length-1){
               index=-1
           }
           index++
           preSum+=code[i]
           res.push(preSum)
       }
    }
    return res
};

消耗时间和内存情况:

方法二 滑动窗口

无论k大于0还是小于0,其实都是一个长度为k的绝对值的窗口,窗口往右移动,每次去掉窗口的第一个值,加入窗口的后一个值,当窗口的左边界或右边界到达数组末尾就要跳转到数组开头,每次窗口的和就是一个密码值

javascript 复制代码
var decrypt = function(code, k) {
    if(k===0) return new Array(code.length).fill(0)
    let res=[],start=0,end=0,preSum=0
    if(k>0){
       start=0,end=k-1
    }else{
       start=code.length+k-1,end=start-k-1
    }
    for(let i=start;i<=end;i++){
        preSum+=code[i]
    }
    for(let i=0;i<code.length;i++){
        preSum-=code[start]
        start++
        if(start==code.length) start=0

        end++
        if(end==code.length) end=0
        preSum+=code[end]

        res.push(preSum)
    }
    return res
};

消耗时间和内存情况:

相关推荐
儿创社ErChaungClub5 分钟前
解锁编程新境界:GitHub Copilot 让效率翻倍
人工智能·算法
前端西瓜哥9 分钟前
贝塞尔曲线算法:求贝塞尔曲线和直线的交点
前端·算法
小灰灰爱代码17 分钟前
C++——求3个数中最大的数(分别考虑整数、双精度数、长整数的情况),用函数模板来实现。
开发语言·c++·算法
南加第一划水20 分钟前
Leetcode 每日一题:Evaluate Division
算法·leetcode·职场和发展
好名字082128 分钟前
monorepo基础搭建教程(从0到1 pnpm+monorepo+vue)
前端·javascript
c#上位机1 小时前
C#事件的用法
java·javascript·c#
chnyi6_ya1 小时前
一些写leetcode的笔记
笔记·leetcode·c#
逝去的秋风1 小时前
【代码随想录训练营第42期 Day61打卡 - 图论Part11 - Floyd 算法与A * 算法
算法·图论·floyd 算法·a -star算法
万物得其道者成1 小时前
React Zustand状态管理库的使用
开发语言·javascript·ecmascript
小白小白从不日白1 小时前
react hooks--useReducer
前端·javascript·react.js