leetcode 43. 字符串相乘
给定两个以字符串形式表示的非负整数 num1 和 num2,返回 num1 和 num2 的乘积,它们的乘积也表示为字符串形式。
示例 1:
输入: num1 = "2", num2 = "3"
输出: "6"
示例 2:
输入: num1 = "123", num2 = "456"
输出: "56088"
说明:
num1 和 num2 的长度小于110。
num1 和 num2 只包含数字 0-9。
num1 和 num2 均不以零开头,除非是数字 0 本身。
不能使用任何标准库的大数类型(比如 BigInteger)或直接将输入转换为整数来处理。
思路
/**
num1的第i位(高位从0开始)和num2的第j位相乘的结果在乘积中的位置是[i+j, i+j+1]
例: 123 * 45, 123的第1位 2 和45的第0位 4 乘积 08 存放在结果的第[1, 2]位中
index: 0 1 2 3 4
1 2 3
* 4 5
---------
1 5
1 0
0 5
---------
0 6 1 5
1 2
0 8
0 4
---------
0 5 5 3 5
这样我们就可以单独都对每一位进行相乘计算把结果存入相应的index中
**/
num1[i]
和num2[j]
的乘积对应的就是res[i+j]
和res[i+j+1]
这两个位置。
索引从左往右,结果位数最大为m+n
,相乘时从后往前,也就是索引从高到底开始遍历
class Solution:
def multiply(self, num1: str, num2: str) -> str:
m = len(num1)
n = len(num2)
res = [0] * (m + n) # 初始化结果数组,最长为两个字符串的长度和
for i in range(m - 1, -1, -1): # 从低位开始相乘,最低位的索引最大
for j in range(n - 1, -1, -1):
mul = int(num1[i]) * int(num2[j]) # 乘积
p1 = i + j # 结果的位置(进位)
p2 = i + j + 1 # 结果的位置(当前位)
s = mul + res[p2] # 累加当前位的值
res[p2] = s % 10 # 求余数,更新当前位置的值
res[p1] += s // 10 # 累加进位
i = 0 # 结果开始位置
while i < m + n and res[i] == 0: # 剔除前面无效的0
i += 1
ans = ''
for s in res[i:]: # 非0元素即为结果,转为字符串即可
ans += str(s)
return ans if ans != '' else '0' # 如果为空,说明结果为0