题目:
给定一个由 整数 组成的 非空 数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储单个数字。
你可以假设除了整数 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
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/plus-one
题解:
C
逆序遍历数组,如果最后一位不为9,则最后一位加一并返回该数组;如果最后一位是9,加一后需要进位,最后一位改为0,继续向前遍历,前一位为9则令该位也为0,直到某一位不为9,则不需要继续遍历。如果每一位都是9,加一后结果会多出一位,则需要再构建一个数组,返回新数组。
/**
* Note: The returned array must be malloced, assume caller calls free().
*/
int* plusOne(int* digits, int digitsSize, int* returnSize)
{
int *result=(int *)malloc(sizeof(int)*(digitsSize+1));
for(int i=digitsSize-1;i>=0;i--)
{
if(digits[i]==9) digits[i]=0;
else
{
digits[i]+=1;
*returnSize=digitsSize;
return digits;;
}
}
*returnSize=digitsSize+1;
result[0]=1;
for(int i=0;i<digitsSize;i++)
{
result[i+1]=digits[i];
}
return result;
}