给你两个二进制字符串,返回它们的和(用二进制表示)。
输入为 非空 字符串且只包含数字 1 和 0。
示例 1:
输入: a = "11", b = "1"
输出: "100"
示例 2:
输入: a = "1010", b = "1011"
输出: "10101"
提示:
每个字符串仅由字符 '0' 或 '1' 组成。
1 <= a.length, b.length <= 10^4
字符串如果不是 "0" ,就都不含前导零。
class Solution:
def addBinary(self, a: str, b: str) -> str:
res = '' # 返回结果字符串
remain = 0 # 保存进位
i = len(a) - 1
j = len(b) - 1
while(i>=0 and j>=0): # 一直计算到较短的字符串结束
if int(a[i]) + int(b[j]) + remain >= 2:
res = str(int(a[i]) + int(b[j]) + remain - 2) + res
remain = 1
i -= 1
j -=1
else:
res = str(int(a[i]) + int(b[j]) + remain) + res
remain = 0
i -= 1
j -=1
if len(a) == len(b): # 如果a和b字符串一样长
if remain == 1: # 且进位为1, 则结果前面再写一个1
res = '1' + res
elif len(a) > len(b):# 如果a长度大于b
i = len(a) - len(b) - 1
while(i >= 0): # 一直遍历到较长的剩下的字符串计算完
if int(a[i]) + remain >= 2:
res = str(int(a[i]) + remain - 2) + res
i -= 1
else:
res = str(int(a[i]) + remain) + res
remain = 0
i -= 1
if remain == 1:
res = '1' + res
else:
i = len(b) - len(a) - 1
while(i >= 0):
if int(b[i]) + remain >= 2:
res = str(int(b[i]) + remain - 2) + res
i -= 1
else:
res = str(int(b[i]) + remain) + res
remain = 0
i -= 1
if remain == 1:
res = '1' + res
return res