Leetcode 66 加一 C语言

Leetcode 66 加一 C语言

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

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

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

示例 1:

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

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

思路1：2年前写的。。先判断如果数字都是9的话，则会产生进位；carry用于表示是否需要多一位，p用于表示进位；先分配内存，从后往前利用carry的距离差进行赋值。

int* plusOne(int* digits, int digitsSize, int* returnSize) {
    int carry = 1;
    for (int i = 0; i < digitsSize; i++){
        if (digits[i] != 9){
            carry = 0;
            break;
        }
    }
    int* plusone = (int*)malloc(sizeof(int)*(digitsSize+carry));
    int p = 1;    
    for (int i = digitsSize-1 ; i >= 0; i--){  
        plusone[i+carry] = digits[i] + p;
        if(plusone[i+carry]>=10)
        {
            plusone[i+carry]-=10;
            p=1;
        }
        else
        {
            p=0;
        }
    }
    if(carry)
        plusone[0]=1;
  *returnSize = digitsSize+carry;
    return plusone;
}

思路2：从数组尾部开始模拟加法运算；并在原数组上修改，若最高位无进位，则直接返回；若有进位，则需申请空间，扩充原数组。

int* plusOne(int* digits, int digitsSize, int* returnSize){
   int flag=1;
   for(int i=digitsSize-1;i>=0;i--)
   {
       if(digits[i]==9&&flag==1)
       {
             digits[i]=0;
             flag=1;
       }
       else if(digits[i]!=9)
       {
           digits[i]=digits[i]+flag;
           flag=0;
       }
   }
   *returnSize=digitsSize+flag;
   if(*digits!=0)
   {
      // *returnSize=digitsSize;
       return digits;
   }
   else
   {
       //*returnSize=digitsSize+1;
       int *p2=(int *)malloc(sizeof(int)*(*returnSize));
       memset(p2,0,*returnSize);
       p2[0]=1;
       for(int j=1;j<*returnSize;j++)
       {
           p2[j]=digits[j-1];
       }
       return p2;
   }
   return 0;
}