《LeetCode之每日一题》:101.加一

Posted 2021-08-26 是七喜呀！

加一

• 有关题目
• • 题解

题目链接： 加一

有关题目

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

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

你可以假设除了整数 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。

代码一：

/**
 * Note: The returned array must be malloced, assume caller calls free().
 */
int* plusOne(int* digits, int digitsSize, int* returnSize){
    int i = digitsSize - 1;
    for (; i >= 0; i--)
    {
        if (digits[i] == 9)
      {
        digits[i] = 0;//下一步直接返会digits
      }
        else
      {
        digits[i] += 1;
        *returnSize = digitsSize;
        return digits;
      }
    }
    
    *returnSize = digitsSize + 1;
    //如果所有位都是进位，则长度+1
    int* rec = (int*)malloc((digitsSize + 1) * sizeof(int));
    memset(rec,0,(digitsSize + 1) * sizeof(int));
    rec[0] = 1;
    return rec;
}

代码二：

/**
 * Note: The returned array must be malloced, assume caller calls free().
 */
int* plusOne(int* digits, int digitsSize, int* returnSize){
	//189，99，123例子
    *returnSize = digitsSize;
    int i = digitsSize - 1;
    for (; i >= 0; i--)
    {
        digits[i]++;
        digits[i] %= 10;
        if (digits[i] != 0) return digits;
    }

    *returnSize = digitsSize + 1;
    //如果所有位都是进位，则长度+1
    int* rec = (int*)malloc((digitsSize + 1) * sizeof(int));
    memset(rec,0,(digitsSize + 1) * sizeof(int));
    rec[0] = 1;
    return rec;
}

时间复杂度：O(n)
空间复杂度：O(1)，不考虑返回数组的大小