66. 加一 - 力扣(LeetCode)
思路:
数字加法应该从前向后遍历,因此我们应该从数字末尾从后向前遍历。如果数字不为9,则直接在末尾 1。如果末尾为9,那么将其变为0,并且在下一位 1。如果一直遍历都为1,那么在数组第0位插入1。
我们在对数字进行 1时,只需要关注数组的末尾有多少个9即可。
如果末尾没有9,那么直接在末尾 1。
如果有若干个9,那么我们只需要在末尾找到第一个不为9的元素。随后将末尾的9全部置为0。
如果所有元素都是9,那么我们只需要构造出一个比原数组多1个长度的数组,将首元素置为1,其余置为0即可。
代码语言:javascript复制/**
* Note: The returned array must be malloced, assume caller calls free().
*/
int* plusOne(int* digits, int digitsSize, int* returnSize) {
for(int i=digitsSize-1;i>=0;--i)
{
digits[i]=digits[i] 1;//判断元素 1是不是10
if(digits[i]!=10)
{
*returnSize=digitsSize;
//如果当前位不为9,那么返回数组的长度就是当前数组的长度
return digits;
}
if(digits[i]==10)
{
digits[i]=0;//如果当前位为9,那么把当前位置为0
}
}
//如果元素全为9,那么开辟新的数组
int*ans=malloc(sizeof(int)*(digitsSize 1));
memset(ans,0,sizeof(int)*(digitsSize 1));
ans[0]=1;
*returnSize=digitsSize 1;
return ans;
}