Skip to content

【力扣】66.加一 这是一道让人容易马失前蹄的算法题

Offer 驾到,掘友接招!我正在参与2022春招打卡活动,点击查看活动详情

春招打卡第22天第32篇。

勤学似春起之苗,不见其增,日有所长;辍学如磨刀之石,不见其损,日有所亏。

掘金的活动真多哇,这个月决定每天用go刷题,一方面提升一下算法水平,另一方面沉淀一下go语言的学习。

Let's GO!

题目描述

给定一个由 整数 组成的 非空 数组所表示的非负整数,在该数的基础上加一。

最高位数字存放在数组的首位, 数组中每个元素只存储单个数字。

你可以假设除了整数 0 之外,这个整数不会以零开头。

示例

示例 1:

输入:digits = [1,2,3]

输出:[1,2,4]

解释:输入数组表示数字 123。

示例 2:

输入:digits = [4,3,2,1]

输出:[4,3,2,2]

解释:输入数组表示数字 4321。

示例 3:

输入:digits = [0]

输出:[1]

提示:

1 <= digits.length <= 100

0 <= digits[i] <= 9

题目分析

很简单哇,获得数组的最后一个元素,进行加一处理,再拼接回去就行啦。

(正如题目所说:这是一道容易让人马失前蹄的算法题)

才没这么简单呢!我们还要考虑最后一位为9的情况、最后几位为9的情况和全部是9的情况。

思路讲解

总结归纳一下,其实就以下这三种情况:

  1. 如果数组的末尾没有 9,例如 [1, 2, 3],那么我们直接将末尾的数加一,得到 [1, 2, 4] 就ok了;

  2. 如果数组的末尾有若干个 9,例如 [3, 2, 1, 9, 9],我们需要找出从末尾开始的第一个不为 9 的元素,也就是1,将该元素加一,得到 [3, 2, 1, 9, 9]。然后将末尾的 9 全部置零,得到 [3, 2, 2, 0, 0]就ok了。

  3. 如果数组的所有元素都是 9,例如 [9, 9, 9],那么答案为 [1, 0, 0, 0]。这种情况下需要构造一个长度比输入数组大1的新数组,将首元素设置为1,其余元素置为0就ok了。

AC代码

go
func plusOne(digits []int) []int {     //获得输入数组