67. 二进制求和python

67. 二进制求和python

问题描述

给你两个二进制字符串，返回它们的和（用二进制表示）。
输入为 非空 字符串且只包含数字 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(object):
    def addBinary(self, a, b):
        """
        :type a: str
        :type b: str
        :rtype: str
        """
        result = []
        a = list(a)
        b = list(b)
        a.reverse()
        b.reverse()
        if len(b) > len(a):
            t = a
            a = b
            b = t
        index = 0
        add = 0
        # 正常每位相加
        while index < len(a) and index < len(b):
            t = int(a[index]) + int(b[index]) + add
            result.append(str(t % 2))
            add = t // 2
            index += 1
        # 如果a还有长度并且有进位，进位数值还是要和a相加一次
        while add != 0 and index < len(a):
            t = int(a[index]) + add
            result.append(str(t % 2))
            add = t // 2
            index += 1
        # 说明当前a的没有需要相加的位数，但是还有一个进位增加最后的结果长度
        if add != 0:
            result.append(str(add))
        # 说明当前a只有剩余直接放到最后结果中
        if index < len(a):
            result.extend(a[index:len(a)])
        result.reverse()
        return "".join(result)

心得

最关键的是a,b两个串当一个串加完了，如果有进位的处理，要考虑到进位有可能和剩余的串继续相加，要是没有剩余串，进位增加最后的结果长度