LeetCode43,一题让你学会高精度算法
本文始发于个人公众号:TechFlow,原创不易,求个关注
今天是LeetCode系列第22篇文章,今天讲的内容是高精度算法。
今天和大家讨论的算法是高精度,对应的LeetCode是第43题。题面其实没什么好说的,以字符串的形式给定两个数字,要求返回这两个数字的乘积。之所以是以字符串的形式给数字是因为这个数字可能会非常大,题目当中给定的范围是110位的数字。对于Python来说这不是问题,但是对于C++和Java等语言来说这么大的数字是无法以int类型存储的,所以必须要使用字符串来接收。
如果你使用Python,你可以不用任何算法就AC这题,但是这没有任何意义。那么正确的方法应该怎么做呢?
高精度与打竖式
这就需要我们的高精度算法出场了,其实严格说起来高精度并不是一种算法,而是一种思想。这个思想非常朴素,我敢保证我们每一个人都学过。还记得小学的时候,我们计算多位数的乘法是怎么算的吗?大家应该都不陌生才对,就是打竖式,like this:
我们人类要打竖式是因为我们只能计算一位数以内的加减乘除,超过一位的人脑不能直接计算,我们就需要用纸笔记录下来进行计算。
纸笔计算的方法很简单,就是一位一位地计算,用每一位数字依次去计算乘法,最后再移位相加起来就得到结果了。
比如在上图的第一个例子当中,我们要计算15 * 16,我们先计算6 * 15的结果,再计算1 * 15,最后将两个结果错位相加,就得到了答案。我们要错位的原因也很简单,因为我们在计算15 * 1的时候,其实背后代表的是15 * 10。我们继续拆分问题,当我们计算6和15相乘的时候,又是怎么计算的呢?顺着这个思路,整个过程可以进一步被划分成先计算6和5相乘,再计算6和1相乘。
最后,我们把两个较大数字的相乘拆分成了在每一位上的数字相乘。到了这里,剩下的就简单了,也就是说我们可以把这两个很大的数字用两个数组来存储,数组当中的每一位存储数字上的一位。
比如我们要计算123 * 224, 我们的第一个数组是[1, 2, 3],我们的第二个数组是[2, 2, 4]。我们仿照乘法竖式中的方法计算这两个数组当中两两的乘积,并将它们拼装成答案。
1 2 3
* 2 2 4
____________
4 9 2
2 4 6
2 4 6
____________
2 7 5 5 2
同样我们用数组来存储中间和最后的结果,最后的结果就是:[2, 7, 5, 5, 2]。由于题目需要我们要返回的是字符串,所以我们还需要将数组里的内容再拼接成字符串。
这种用数组来模拟数字进行加减乘除运算的方法就叫做高精度算法,相信大家也都看到了,严格说起来这并不是一个算法,而只是一种思想。今天的题目出的是乘法,我们利用同样的方法也可以计算加减和除法。其中加减法非常简单,而除法则要复杂得多,也是高精度当中最难实现的部分。这里我们不做过多的拓展,计算的方法同样是打竖式,感兴趣的同学可以自行实现。
进位和前导零
当我们理清楚了打竖式的方法之后,我们还要面临进位和前导零的问题。
进位应该很容易理解,我们需要在计算乘法的时候判断当前位置的元素是否大于等于10,如果超过10的话,我们则需要进行进位。我们只需要将它除以10,得到的结果就是我们需要进位的值。除此之外就是前导零的问题,我们都知道除了零以外的合法数字是不允许首位出现0的,但是由于我们计算的是乘法,所以当其中某一个数为0会得到整体的结果为0,但是表示在数组当中则是多个0.
举个简单的例子,比如123 * 0,最后得到的应该是0,但是由于我们用数组表示了乘法运算当中的每一位,并且还进行了加法计算,所以会导致出现000的结果。这种情况我们要做特殊的处理,不过这也不复杂。最后我们把上面所有的思路都整理一下,就可以得到结果了。
我们来看下代码:
class Solution:
def multiply(self, num1: str, num2: str) -> str:
# 将字符串转化成数组
# 翻转数组,因为我们用第0位表示个位
arr1 = [ord(i) - ord('0') for i in num1][:: -1]
arr2 = [ord(i) - ord('0') for i in num2][:: -1]
# 创建结果数组,可以证明结果的长度最多是n + m
n, m = len(arr1), len(arr2)
ret = [0 for i in range(n + m + 1)]
for i in range(n):
for j in range(m):
# 按位相乘,计算进位
ret[i + j] += arr1[i] * arr2[j]
if ret[i+j] >= 10:
ret[i+j+1] += ret[i+j] // 10
ret[i+j] %= 10
# 最后把数组再转化成字符串返回
# 去除前导零
result = ''.join(map(str, ret))[::-1].lstrip('0')
return result if len(result) > 0 else '0'
今天的题只是Medium难度,并不算困难,会选这题的原因主要是为了高精度算法。高精度算法本身并不难,也并不常用即使是在算法比赛当中也不常见。但是它给了我们一个思路,当我们要计算的数值超过计算机目前承载能力的时候,我们还有什么方法?
当然这题我们也可以取巧,因为Python当中内置了大整数,当它检测到我们的计算结果超过范围的时候,会自动转化成大整数来进行计算。所以这题如果我们使用Python,可以只用几行代码搞定:
class Solution:
def multiply(self, num1: str, num2: str) -> str:
num1 = int(num1)
num2 = int(num2)
return str(num1 * num2)
今天关于高精度算法的内容就到这里,如果觉得有所收获,请顺手点个关注或者转发吧,你们的举手之劳对我来说很重要。