给定一个由 整数 组成的 非空 数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储单个数字。
你可以假设除了整数 0 之外,这个整数不会以零开头。
加一问题重点在于有无进位,所以需要先遍历9的个数。然后分情况,当9的个数小于元素个数时,需将是9的元素变为0,然后在最左边的9的前一位加一进位。当没有9时,那直接将最右边的元素加一即可。当数组全是9时,因为进位导致数组溢出,所以重新开辟空间。将第零个元素置1,其他为0即可。
/**
* Note: The returned array must be malloced, assume caller calls free().
*/
int* plusOne(int* digits, int digitsSize, int* returnSize){
int i=digitsSize-1;
int count=0;
while(digits[i]==9 && count<digitsSize) //遍历数组记录9的个数
{
if(i>0) //leetcode中的数组下标不能减到小于零,会报错,所以用count记录
i--;
count++;
}
if(count<digitsSize &&count>0) //9的个数小于数组元素个数
{
int j=0;
for(j=digitsSize-1;j>=(digitsSize-count);j--) //将是9的都进位变成零,然后前面一位加一
{
digits[j]=0;
}
digits[j]+=1;
}
else if(count==0) //9的个数等于零,直接将最小位的值加一
{
digits[digitsSize-1]+=1;
}
else if(count==digitsSize) //当数组元素都是9时
{
int *arr=(int*)malloc((digitsSize+1)*sizeof(int)); //因为全是9会进位,空间不足,需要重新开辟数组
memset(arr, 0, sizeof(int) * (digitsSize + 1));
arr[0]=1;
*returnSize=digitsSize+1;
return arr;
}
*returnSize=digitsSize;
return digits;
}