给定一个由 整数 组成的 非空 数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储单个数字。
你可以假设除了整数 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]
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/plus-one
解题思路:可以为情况讨论
1)当数组元素最后一个不为9时,让最后一位加一,直接将原数组放到新定义的数组中,再返回出来
2)当数组里面的元素全为9时,让数组长度加一,让新数组的第一位为1,其余全为0
3)当数组里面元素不全为9时,---->见代码部分
/**
* Note: The returned array must be malloced, assume caller calls free().
*/
int* plusOne(int* digits, int digitsSize, int* returnSize){
int *a=(int *)malloc(sizeof(int)*(digitsSize+1));
int i,cnt=0;
if(digits[digitsSize-1]!=9){ //当最后一位数不为9时
*returnSize=digitsSize;
a[digitsSize-1]=digits[digitsSize-1]+1;//直接让最后一位加一
for(i=0;i<digitsSize-1;i++){
a[i]=digits[i];//其他直接等过去
}
return a;
}
else{
for(i=0;i<digitsSize;i++){
if(digits[i]==9){//判断数组里面有多少个9
cnt++;
}
}
if(digitsSize==cnt){//如果数组长度等于cnt,代表数组里的数全为9
*returnSize=digitsSize+1;
a[0]=1;//第一位置为一
for(i=1;i<(digitsSize+1);i++){
a[i]=0;//其余位置全为0
}
return a;
}else{//当9的个数小于digitsSize时
int t;
digits[digitsSize-1]+=1;//让最后一位数加1
t=digits[digitsSize-1];//让t等于数组的后一位数
a[digitsSize-1]=t%10;// 新数组a的最后一位对t取余
for(i=digitsSize-2;i>=0;i--){
a[i]=(digits[i]+t/10)%10;//新数组a让原数组(digits)加上后一位数除以10,防止a[i]大于10需要取余
t=digits[i]+t/10;//t为a[i]取余前的数
}
*returnSize=digitsSize;
return a;
}
}
return a;
}
注意:每次确定好数组长度要用*returnSize等于该长度