题目:
给定一个由 整数 组成的 非空 数组所表示的非负整数,在该数的基础上加一。最高位数字存放在数组的首位, 数组中每个元素只存储单个数字。你可以假设除了整数 0 之外,这个整数不会以零开头。
思路:
因为是在该数的基础上+1,那么即使有进位,进位也为1。基于这个的前提下,我们可以分为3中想法进行思考:
这个数组不会有9,那么就是正常末尾+1
这个数组内部部分有9,那么就是是9的那位变为0,前一位+1
这个数组内部全部为9,那么构造一个新数组,第一位为1,其他全为0
class Solution {
public int[] plusOne(int[] digits) {
int right = digits.length-1;
for(int i = digits.length-1;i>=0;i--){
if(digits[i]!=9){
digits[i]+=1;
return digits;
}else{
digits[i]=0;
}
}
int[] arr = new int[digits.length+1];
arr[0]=1;
for(int i=1;i<arr.length;i++){
arr[i]=0;
}
return arr;
}
}
我翻了一下力扣的评论区,想找一个更简单的方法,有一个朋友用栈来解决这个问题。栈是先进后出的,你将数组压入进去,正好最后一位是末尾,然后进行+1处理。
class Solution {
public int[] plusOne(int[] digits) {
Deque<Integer> stack1 = new ArrayDeque<>();
Deque<Integer> stack2 = new ArrayDeque<>();
for (int i:digits
) {
//将数组中的各个数放入栈1中
stack1.push(i);
}
int m = stack1.size();
//栈1不为空
while (!stack1.isEmpty()){
int temp = stack1.pop();
if(temp!=9){
stack2.push(temp+1);
break;
}else{
stack2.push(0);
if(stack1.isEmpty()){
stack1.push(0);
}
}
}
int[] arr;
if(m<stack2.size()){
arr= new int[digits.length+1];
}else {
arr = new int[digits.length];
while(!stack1.isEmpty()){
stack2.push(stack1.pop());
}
}
for(int i = 0; !stack2.isEmpty(); i++){
arr[i]=stack2.pop();
}
return arr;
}
}