日拱一卒,月进一步(4)

2024-05-08 08:39:37 浏览数 (1)

66. 加一 - 力扣(LeetCode)

思路:

数字加法应该从前向后遍历,因此我们应该从数字末尾从后向前遍历。如果数字不为9,则直接在末尾 1。如果末尾为9,那么将其变为0,并且在下一位 1。如果一直遍历都为1,那么在数组第0位插入1。

我们在对数字进行 1时,只需要关注数组的末尾有多少个9即可。

如果末尾没有9,那么直接在末尾 1。

如果有若干个9,那么我们只需要在末尾找到第一个不为9的元素。随后将末尾的9全部置为0。

如果所有元素都是9,那么我们只需要构造出一个比原数组多1个长度的数组,将首元素置为1,其余置为0即可。

代码语言:javascript复制
/**
 * 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)
    {
        digits[i]=digits[i] 1;//判断元素 1是不是10
        if(digits[i]!=10)
        {
            *returnSize=digitsSize;
            //如果当前位不为9,那么返回数组的长度就是当前数组的长度
            return digits;
        }
        if(digits[i]==10)
        {
            digits[i]=0;//如果当前位为9,那么把当前位置为0
        }
    }
    //如果元素全为9,那么开辟新的数组
    int*ans=malloc(sizeof(int)*(digitsSize 1));
    memset(ans,0,sizeof(int)*(digitsSize 1));
    ans[0]=1;
    *returnSize=digitsSize 1;
    return ans;
}

0 人点赞