1. 题目
给定一个由 整数 组成的 非空 数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储单个数字。
你可以假设除了整数 0 之外,这个整数不会以零开头。
2. 示例
示例 1:
输入:digits = [1,2,3] 输出:[1,2,4] 解释:输入数组表示数字 123。示例 2:
输入:digits = [4,3,2,1] 输出:[4,3,2,2] 解释:输入数组表示数字 4321。示例 3:
输入:digits = [0] 输出:[1]
3. 解决思路及分析
这道题要考虑几种情况,1.若最后一位数字小于9,那么只需给其加1就可返回数组2.若最后一位为9,要考虑进位问题,前一位+1,并继续考虑是否上一位是否为9。此外若数组全为9还需要考虑数组大小溢出问题,需要开辟新的数组,但此时只需给新数组的首位置1即可。
/**
* 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
if(digits[i]!=10) //若不等于10,直接返回数组大小和数组即可。
{
*returnSize=digitsSize; //将数组大小赋给指针
return digits; //返回值
}
if(digits[i]==10) //若最后一位等于10,则该位请0,且检索到的下一位+1,直到检索的位不等于10,则返回数组和大小
{
digits[i]=0;
}
}
int *ans=malloc(sizeof(int)*(digitsSize+1));//若运行到这步,没有返回,说明所有位都进行了+1操作,则数组元素全为9,数组长度发生变化,此时开辟新的数组,长度为原长度+1,首位为1
memset(ans,0,sizeof(int)*(digitsSize+1)); //将新数组清0
ans[0]=1; //全为9的数组+1后首位为1
*returnSize=digitsSize+1; //长度赋给返回指针
return ans; //返回新数组
}