思路:
数字加法应该从前向后遍历,因此我们应该从数字末尾从后向前遍历。如果数字不为9,则直接在末尾+1。如果末尾为9,那么将其变为0,并且在下一位+1。如果一直遍历都为1,那么在数组第0位插入1。
我们在对数字进行+1时,只需要关注数组的末尾有多少个9即可。
如果末尾没有9,那么直接在末尾+1。
如果有若干个9,那么我们只需要在末尾找到第一个不为9的元素。随后将末尾的9全部置为0。
如果所有元素都是9,那么我们只需要构造出一个比原数组多1个长度的数组,将首元素置为1,其余置为0即可。
/** * 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; }