难中见易，别被题名唬住：揭秘LeetCode 66 题“加一”背后的简单精髓

前端

2023-11-05 19:45:02

引言：算法入门之路，从LeetCode开始

LeetCode是一个备受推崇的算法题库，收录了数千道算法题，涵盖各种难度和领域，是程序员们磨炼算法技能的必备工具。作为LeetCode题库中一道经典题目，“加一”以其简洁的题干和丰富的解题技巧，吸引了众多算法爱好者的关注。

题干剖析：理解“加一”的本质

LeetCode 66 题“加一”的题干十分简单：给定一个由整数组成的非空数组，表示一个非负整数，请你在该数的基础上加一。

乍一看，这道题似乎很简单，只需要对数组中的每个数字依次加一即可。然而，事情并非如此简单。当数组中存在进位时，我们需要对进位进行处理，这才是这道题的关键所在。

解题思路：巧用数组特性，化繁为简

解决这道题的关键在于理解数组的特性。在数组中，每个元素都代表一个数字，我们可以通过索引来访问这些元素。同时，数组具有连续性，我们可以利用这个特性来处理进位问题。

具体来说，我们可以从数组的最后一个元素开始，对每个元素依次加一。如果加一后该元素的值大于或等于10，则需要将该元素的值减去10，并在下一个元素上加上1，以此类推，直到进位处理完成。

代码实现：一步步拆解，轻松掌握

def plus_one(digits):
  """
  给定一个由整数组成的非空数组，在该数的基础上加一。
  最高位数字存放在数组的首位，数组中每个元素只存储单个数字。
  你可以假设除了整数0之外，这个整数不会以零开头。

  Args:
    digits (list): 由整数组成的非空数组

  Returns:
    list: 加一后的数组
  """

  # 从最后一个元素开始，对每个元素依次加一
  for i in range(len(digits) - 1, -1, -1):
    # 如果加一后该元素的值大于或等于10，则需要将该元素的值减去10，并在下一个元素上加上1
    if digits[i] + 1 >= 10:
      digits[i] -= 10
      if i > 0:
        digits[i - 1] += 1
    # 否则，直接将该元素的值加一
    else:
      digits[i] += 1
      break

  # 如果数组的第一个元素也需要进位，则需要在数组的最前面添加一个元素
  if digits[0] == 0:
    digits.insert(0, 1)

  return digits