JavaScript lastElement函数问题求助：代码失效原因与改进建议

嘿，Andreas！我来帮你排查下这个函数的问题，再给你几个优化的思路～

你的代码为什么无效？ #

咱们先拆解下当前代码的问题：

function lastElement (num) { 
  if (num !== undefined){ 
    return num[num.length -1]; 
  } 
  return null; 
}

• 错误的判断逻辑：你只检查了参数是否为undefined，但空数组是一个合法的数组对象（不是undefined），所以当传入空数组时，代码会执行num[num.length -1]。空数组的length是0，0-1=-1，而数组的[-1]在JS里会返回undefined，这不符合题目要求的“空数组返回null”。
• 参数命名不够清晰：num这个名字很容易让人误解成单个数字，但你的参数其实是数组，建议改成arr或者array，能大幅提升代码的可读性。

重构思路 #

这里给你几个不同风格的重构方案，你可以根据自己的习惯选择：

方案1：修复原逻辑，保持清晰易懂 #

这个方案在你的基础上调整判断条件，覆盖空数组和无参数的情况：

function lastElement(arr) {
  // 先判断数组是否存在，且长度大于0
  if (arr && arr.length > 0) {
    return arr[arr.length - 1];
  }
  // 空数组或未传参数时返回null
  return null;
}

方案2：用短路求值简化代码 #

利用JS的逻辑运算符，让代码更简洁：

function lastElement(arr) {
  // 可选链处理参数未传入的情况，length>0时返回最后一个元素，否则返回null
  return arr?.length ? arr[arr.length - 1] : null;
}

方案3：用ES2022+的新特性极简实现 #

如果你的环境支持ES2022及以上，可以用Array.prototype.at()和空值合并运算符，代码超简洁：

function lastElement(arr) {
  // at(-1)直接取最后一个元素，空数组/无参数时返回undefined，再转成null
  return arr?.at(-1) ?? null;
}

测试验证 #

你可以用这些案例测试函数是否符合要求：

• lastElement([1, 2, 3]) → 返回3（正确）
• lastElement([]) → 返回null（正确）
• lastElement() → 返回null（正确）
• lastElement(['apple', 'banana']) → 返回'banana'（正确）

内容的提问来源于stack exchange，提问作者Andreas Fitzner-Haupt