leetcode 66. 加一

给定一个由 整数 组成的 非空 数组所表示的非负整数,在该数的基础上加一。

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

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

leetcode 66. 加一_第1张图片

 加一问题重点在于有无进位,所以需要先遍历9的个数。然后分情况,当9的个数小于元素个数时,需将是9的元素变为0,然后在最左边的9的前一位加一进位。当没有9时,那直接将最右边的元素加一即可。当数组全是9时,因为进位导致数组溢出,所以重新开辟空间。将第零个元素置1,其他为0即可。

/**
 * Note: The returned array must be malloced, assume caller calls free().
 */
int* plusOne(int* digits, int digitsSize, int* returnSize){
        int i=digitsSize-1;
        int count=0;
       
        while(digits[i]==9 && count0)             //leetcode中的数组下标不能减到小于零,会报错,所以用count记录
                 i--;
                count++;
            }
        if(count0)   //9的个数小于数组元素个数
        {
            int j=0;
            for(j=digitsSize-1;j>=(digitsSize-count);j--)  //将是9的都进位变成零,然后前面一位加一
            {
                digits[j]=0;             
            }
            digits[j]+=1;
        }
        else if(count==0)   //9的个数等于零,直接将最小位的值加一
        {
            digits[digitsSize-1]+=1;
        }
        else if(count==digitsSize)   //当数组元素都是9时
        {
            int *arr=(int*)malloc((digitsSize+1)*sizeof(int));  //因为全是9会进位,空间不足,需要重新开辟数组
            memset(arr, 0, sizeof(int) * (digitsSize + 1));    
            arr[0]=1;
            *returnSize=digitsSize+1;
            return arr;
        }
        *returnSize=digitsSize;
        return digits;
}

你可能感兴趣的:(leetcode刷题,leetcode,算法,职场和发展,c语言,开发语言)