LeetCode：第66题 加1 C语言实现 2020.2.4

给定一个由整数组成的非空数组所表示的非负整数，在该数的基础上加一。 最高位数字存放在数组的首位， 数组中每个元素只存储单个数字。 你可以假设除了整数 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 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

java new int[2]默认赋初值为0

 

/*
安安思路：由于题目要求要malloc。所以先malloc了一个和原数组相同大小的数组
如果最高位还有进位，则这个数组不够存储，所以就又新开了一个比这个数组尺寸大1的数组
进位用一个变量carry表示

anan
2020.2.4
C语言
leetcode 66. 加一
*/

/**
 * Note: The returned array must be malloced, assume caller calls free().
 */
int* plusOne(int* digits, int digitsSize, int* returnSize){
    int *p;
    int carry = 0;

    p = (int *)malloc(sizeof(int)*digitsSize);
    for(int i = digitsSize-1; i >= 0; i--){
        //printf("%d ", digits[i]);
        p[i] = digits[i] + carry;
        if(i == digitsSize-1){
            p[i]+=1;
        }
        if(p[i] == 10){
            p[i] = 0;
            carry = 1;
        }else{
            carry = 0;
        }
    }
    printf("\n");

    if(carry == 1){   //表示最高位还有进位
        *returnSize = digitsSize+1;
        int *q = (int *)malloc(sizeof(int)*(digitsSize+1));
        for(int i = 0; i < digitsSize; i++){
            q[i+1] = p[i];
        }
        q[0] = 1;
        return q;   
    }else{
        *returnSize = digitsSize;
        return p;
    }
    

}

看了一些别人的题解之后，发现自己题解的问题

1.当加1之后位数不变时，不需要新开数组，只需要在原数组上加1，直接返回原数组就好了

2.当加1之后位数改变时，其实就是99，999这种情况，这种情况下，最高位为1，其余位都为0

/*
看了题解之后，自己重写的代码
c语言
*/

/**
 * 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--){
        //printf("%d ", digits[i]);
        if(digits[i] < 9){
            digits[i]++;
            *returnSize = digitsSize;
            return digits;
        }
        digits[i] = 0;
    } 

    *returnSize = digitsSize+1;
    int *p = (int *)malloc(sizeof(int)*(*returnSize));
    p[0] = 1;
    for(int i = 1; i < *returnSize; i++){
        p[i] = 0;
    }
    return p; 
}

//别人的代码 java
//java new int[2]默认赋初值为0

class Solution {
    public int[] plusOne(int[] digits) {
        for (int i = digits.length - 1; i >= 0; i--) {
            digits[i]++;
            digits[i] = digits[i] % 10;
            if (digits[i] != 0) return digits;
        }
        digits = new int[digits.length + 1];
        digits[0] = 1;
        return digits;
    }
}