加一算法解析

package test.java.first;

import org.junit.Test;

import java.util.Stack;
import java.util.concurrent.BlockingDeque;
import java.util.concurrent.LinkedBlockingDeque;

/**
 * 给定一个由整数组成的非空数组所表示的非负整数,在该数的基础上加一。
 *
 * 最高位数字存放在数组的首位, 数组中每个元素只存储单个数字。
 *
 * 你可以假设除了整数 0 之外,这个整数不会以零开头。
 *
 * 示例 1:
 *
 * 输入: [1,2,3]
 * 输出: [1,2,4]
 * 解释: 输入数组表示数字 123。
 * 示例 2:
 *
 * 输入: [4,3,2,1]
 * 输出: [4,3,2,2]
 * 解释: 输入数组表示数字 4321。
 *
 * 来源:力扣(LeetCode)
 * 链接:https://leetcode-cn.com/problems/plus-one
 */
public class Second {

    @Test
    public void secondTest(){
        int[] digits=new int[]{9};
        long start=System.currentTimeMillis();
        plusOne(digits);
        System.out.println(System.currentTimeMillis()-start);
    }
    public int[] plusOne(int[] digits) {
        Stack stack=new Stack();
        BlockingDeque deque=new LinkedBlockingDeque<>();
        for (int i = 0; i < digits.length; i++) {
            stack.push(digits[i]);
        }
        judge(stack,deque,true);
        int[] returnInts=new int[deque.size()];
        for (int i = 0; i < returnInts.length; i++) {
            returnInts[i]=deque.pop();
        }
        return returnInts;
    }
    public void judge(Stack stack,BlockingDeque deque,boolean plus){
        if (!stack.empty()){
            Integer intT=stack.pop();
            int temp=0;
            if (plus){
                temp=intT+1;
            }else {
                temp=intT;
            }
            if (temp>9){
                deque.push(0);
                if (stack.empty()){
                    deque.push(1);
                }
                judge(stack,deque,true);
            }else {
                deque.push(temp);
                judge(stack,deque,false);
            }
        }
    }
}

 

你可能感兴趣的:(算法学习解析)