难度:simple
本题同时也是《剑指offer 专项突击》的第二题。
problem
https://leetcode.cn/problems/add-binary/
以及
https://leetcode.cn/problems/JFETK5/
给定两个 01 字符串 a 和 b ,请计算它们的和,并以二进制字符串的形式输出。
输入为 非空 字符串且只包含数字 1 和 0。
示例 1:
输入: a = “11”, b = “10” 输出: “101”
示例 2:输入: a = “1010”, b = “1011” 输出: “10101”
提示:
每个字符串仅由字符 ‘0’ 或 ‘1’ 组成。
1 <= a.length, b.length <= 10^4
字符串如果不是 “0” ,就都不含前导零。
解法
这题还是比较容易就想到了的。有些解法是将字符串先全部转为十进制int整数,进行整数加法后,再十进制转二进制。这样做可能容易带来溢出的问题。可以直接考虑指针从最末位开始往前移,逢二进一。有的做法是采用双指针,写while。我的直观想法是将短的字符串前面补‘0’,这样使a,b
两个字符串长度相等,就能直接用一个指针从最末位开始往前运算。为此,先将a,b
中长的字符串放于a
,短的给b
。再给b
补前置‘0’。
进入循环后,先对每一位按位相加,超过2的,将进位符c
赋值1,同时用digit_a+digit_b+c-2
更新当前位(我是把它放在列表最后,需要输出时再将列表反转并转为字符串,其实也可以直接字符串后 + 字符串,反转时a[::-1]
即可)。如果不需要进位,将进位符c
赋值0. 遍历完所有位后结束循环,如果此时还有进位,还要在最前面补‘1’。
a=‘111’
b=‘111’
完整代码
class Solution:
def addBinary(self, a: str, b: str) -> str:
la=len(a)
lb = len(b)
if la>=lb:
pass
else:
swap = a
a=b
b=swap
la = len(a)
lb = len(b)
res=[]
c =0
b='0'*(len(a)-len(b))+b
for i in range(la-1, -1, -1):
digit_a = int(a[i])
digit_b = int(b[i])
if digit_a + digit_b + c >=2:
res.append(str(digit_a+digit_b+c-2))
c=1
else:
res.append(str(digit_a+digit_b+c))
c=0
if c==1:
res.append('1')
res.reverse()
res=''.join(res)
return res