Leetcode 题解(每日打卡)
[66.加一]
题目描述
给定一个由 整数 组成的 非空 数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储单个数字。
你可以假设除了整数 0 之外,这个整数不会以零开头。
思路
将数组从后往前遍历,若取出的元素为9的话,则令该位置的值为0,继续遍历,若当i>0时,该值不为9,则该值加1,并返回数组,否则创建新数组,并令数组第0个元素为1,其余元素均为0。
其实该题总共就只有三种情况:
- 不进位
- 进位,但是进位后,数组的总长度不变
- 进位,进位后,数组的总长度加1(全为9的情况)
代码实现
int* plusOne(int* digits, int digitsSize, int* returnSize){
for(int i = digitsSize-1;i >= 0;i--){
if(digits[i] == 9){
digits[i] = 0;
if(i == 0){
*returnSize = digitsSize+1;
}
}else{
digits[i] = digits[i]+1;
*returnSize = digitsSize;
return digits;
}
}
int* arr = (int*)malloc(sizeof(int)*(*returnSize));
arr[0] = 1;
for(int i = 1;i < *returnSize;i++){
*(arr+i) = 0;
}
return arr;
}