在leetcode上有这样一道题,要求写一段代码实现数组的加法,要求如下:
题目链接:力扣
乍一看这不是非常简单嘛,将数组K还原成一个整数,然后与X相加,再把得到的结果按位拆成数组返回,就得到了结果。这样做的确可以通过一部分测试,但是当数组K的元素特别多时,有可能会超过int的大小(2^31-1),此时就无法得到正确的结果。
这个问题解决的是大数相加问题,当两个很大的数相加时,就不能简单的直接相加。采用按位相加就可以避免超出int范围的问题。因此这一道题可以将X的值按位取出,然后和数组K的对应位相加,有进位时就进位,在下一位数相加时加上进位的值(因为每一位的值0~9,所以有进位时进位为1)即可。
代码如下:
/**
* Note: The returned array must be malloced, assume caller calls free().
*/
int* addToArrayForm(int* num, int numSize, int k, int* returnSize){
int ksize = 0;
int knum = k;
while(knum != 0)
{
ksize++;
knum = knum/10;
}//计算K的长度
//相加结果的位数将等于两数位数最大的一个,或者比其大一。分配空间时考虑最大的情况
int len = numSize > ksize ? numSize:ksize;
int * retarr = (int *)malloc(sizeof(int)*(len+1));
int reti = 0;
int nextnum = 0;
while(len--)
{
int a= 0;
if((numSize - 1) >= 0)
{
a = num[numSize-1];
numSize--;
}//每次取一位去进行相加
int sum = a + k%10 + nextnum;
k=k/10;
if(sum >= 10)
{
nextnum = 1;
sum = sum-10;
}
else
{
nextnum = 0;
}
retarr[reti] = sum;
reti++;
}
//计算完毕后检查进位nextnum是否为1,为1还要加上这一位进位
if(nextnum ==1)
{
retarr[reti] = 1;
reti++;
}
//由于存放结果时采用的倒序,所以将数组的内容翻转就可以得到正确结果了
int left = 0,right = reti - 1;
while(left<right)
{
int tmp = retarr[left];
retarr[left] = retarr[right];
retarr[right] = tmp;
left++;
right--;
}
//返回数组的大小等于此时的reti值
* returnSize = reti;
return retarr;
}
写完后提交代码:
完成了题目的所有要求,这道题解答完毕。
这样的写法可能不是最优解,欢迎大家交流指导!