给定一个由整数组成的非空数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储一个数字。
你可以假设除了整数 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 i; int *back = (int *) malloc ((digitsSize + 1) * sizeof(int)); for (i = digitsSize - 1; i >=0; i--) { if (digits[i] < 9) { digits[i]++; break; } else digits[i] = 0; } if (digits[0] == 0) { back[0] = 1; for (int j = 1; j < digitsSize + 1; j++) { back[j] = 0; } * returnSize = digitsSize + 1; return back; } else { *returnSize = digitsSize; return digits; } }