66.加一

一.题目描述

给定一个由 整数 组成的 非空 数组所表示的非负整数，在该数的基础上加一。最高位数字存放在数组的首位， 数组中每个元素只存储单个数字。你可以假设除了整数 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. 末尾数不是9：此时直接将最后一个数加一即可；
2. 末尾数是9或后几位是9，但第一位不是9：此时从后往前遍历，依次将每个遍历到的9变为0，同时将遇到的第一个不为9的数加一；
3. 所有位都是9：此时需要用malloc申请一个比原数组多一的空间，将第一个位置变为1，其他位置为0；

三.代码实现

#include<stdio.h>
#include<stdlib.h>

int* plusOne(int* digits, int digitsSize, int* returnSize) {
    int k = digitsSize-1;
    if (digits[k] != 9)
    {
        digits[k] += 1;
        *returnSize = digitsSize;
    }
    else
    {
        while (digits[k] == 9)
        {
            digits[k] = 0;
            k--;
            if (k == 0 && digits[0] == 9 ||k==-1)
            {
                int* p = (int*)malloc(sizeof(int)*(digitsSize+1));
                p[0] = 1;
                for (int i = 1; i <= digitsSize; i++)
                    p[i] = 0;
                *returnSize += 1;
                //free(p);
                return p;
            }
        }   
        digits[k] += 1;
        *returnSize = digitsSize;
    }
    return digits;
}

int main()
{
    int nums[] = {9};
    int length = sizeof(nums) / sizeof(int);
    int size=length;
    int *str = plusOne(nums, length, &size);
    for (int i = 0; i < size; i++)
        printf("%2d", str[i]);
	return 0;
}