加一

给定一个非负整数组成的非空数组，在该数的基础上加一，返回一个新的数组。

最高位数字存放在数组的首位， 数组中每个元素只存储一个数字。

你可以假设除了整数 0 之外，这个整数不会以零开头。

示例 1:

输入: [1,2,3]
输出: [1,2,4]
解释: 输入数组表示数字 123。

示例 2:

输入: [4,3,2,1]
输出: [4,3,2,2]
解释: 输入数组表示数字 4321。

思路

如果末尾为9，则会进1，不为9，直接加1，需要考虑到全为9时，该数会多一位，只需首位置为1，其余皆为0

实现代码（C语言）

/**
 * Return an array of size *returnSize.
 * Note: The returned array must be malloced, assume caller calls free().
 */
int* plusOne(int* digits, int digitsSize, int* returnSize) {
    
    int i=digitsSize-1;
    int *result=(int *)malloc(sizeof(int)*(digitsSize+1));
    while(i>=0){
        if(digits[i]<9){
             digits[i]=digits[i]+1;
              *returnSize=digitsSize;
              return digits;
        }
             
        digits[i]=0;
        i--;
        
    }
    *returnSize=digitsSize+1;
    result[0]=1;   
    return result;
}

int* plusOne(int* digits, int digitsSize, int* returnSize) {
    
    int count=0;
    int i=digitsSize-1;
    int *result=(int *)malloc(sizeof(int)*(digitsSize+1));
     while(i>=0){
        if(digits[i]==9){
            if(i==0){
                digits[i]=1;
                digits[digitsSize]=0;
                *returnSize=digitsSize+1;
                result=digits;   
                return result;
            }else{
                digits[i]=0;
            
                i--;
            }
           
        }else{
                digits[i]=digits[i]+1;
                *returnSize=digitsSize;
                 return digits;
              
        }
        
    }
    return digits;//根据leetcode机制，必须在这有一个return，虽然执行不到这
}
            

实现代码（Java）

public int[] plusOne(int[] digits) {
        
    int n = digits.length;
    for(int i=n-1; i>=0; i--) {
        if(digits[i] < 9) {
            digits[i]++;
            return digits;
        }
        
        digits[i] = 0;
    }
    
    int[] newNumber = new int [n+1];
    newNumber[0] = 1;
    
    return newNumber;
}