LeetCode：二进制求和

题目

给你两个二进制字符串，返回它们的和（用二进制表示）。

输入为 非空 字符串且只包含数字 1 和 0。

示例 1:

输入: a = “11”, b = “1”
输出: “100”
示例 2:

输入: a = “1010”, b = “1011”
输出: “10101”

解题思路

1、由于两个字符串中的元素都会参与计算，为防止溢出异常，先将短的那个字符串添加‘0’至两个字符串长度相同。
2、如果其中有个为空字符串，则直接返回另一个字符串。
3、加法都是从最右侧的数字开始，因此字符串也得从最后一个字符串开始遍历。
4、用tag表示进位值，每一轮计算都要看a[i]、b[i]、tag的值。

代码实现

class Solution:
    def addBinary(self, a: str, b: str) -> str:
        tag = 0
        result = []
        if not a:
            return b
        elif not b:
            return a
        # 将两者的长度补到一致
        if len(a) > len(b):
            b = ''.join(['0' for i in range(len(a) - len(b))]) + b
        elif len(b) > len(a):
            a = ''.join(['0' for i in range(len(b) - len(a))]) + a
        for i in range(len(max(a,b))-1, -1, -1):
            if a[i] == '1' and b[i] == '1':
                if tag:
                    result.insert(0, '1')
                    tag = 1
                else:
                    result.insert(0, '0')
                    tag = 1
            elif a[i] == '0' and b[i] == '0':
                if tag:
                    result.insert(0, '1')
                    tag = 0
                else:
                    result.insert(0, '0')
                    tag = 0
            else:
                if tag:
                    result.insert(0, '0')
                    tag = 1
                else:
                    result.insert(0, '1')
                    tag = 0
        if tag:
            result.insert(0, '1')
        return ''.join(result)


print(Solution().addBinary("1010", '1011'))