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;
}