leetcode初级数组之加一

给定一个由整数组成的非空数组所表示的非负整数，在该数的基础上加一。
最高位数字存放在数组的首位， 数组中每个元素只存储一个数字。
你可以假设除了整数 0 之外，这个整数不会以零开头。
示例 1:
输入: [1,2,3]
输出: [1,2,4]
解释: 输入数组表示数字 123。
示例 2:
输入: [4,3,2,1]
输出: [4,3,2,2]
解释: 输入数组表示数字 4321。

/**
 * Return an array of size *returnSize.
 * Note: The returned array must be malloced, assume caller calls free().
 */
int* plusOne(int* digits, int digitsSize, int* returnSize) {
    int carry = 1;//表示进位
    int* newarr, tmp;
    int k = digitsSize;
    while(k--)
    {
        //保存新的值，并求新的进位
        tmp = (digits[k] + carry) % 10;
        carry = (digits[k] + carry) / 10;
        digits[k] = tmp;
        if(0 == carry)
        {
            *returnSize = digitsSize;
            return digits;  
        }             
    }
    //运行到这说明最高位进位，申请新数组，并复制原来的值
    newarr = (int*)malloc(sizeof(int)*(digitsSize+1));
    newarr[0] = carry;
    *returnSize = digitsSize + 1;
    while(digitsSize--)
        newarr[digitsSize+1] = digits[digitsSize];
    return newarr;
}