二进制数求和

题目来源：https://leetcode-cn.com/problems/add-binary/

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

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

示例 1:

输入: a = "11", b = "1"
输出: "100"

示例 2:

输入: a = "1010", b = "1011"
输出: "10101"

解法一：循环遍历

设置一个进位标志sign，初始化为0。将a和b转为二进制数后，从最低位开始逐位相加，再加上产生的进位sign。

class Solution(object):
    def addBinary(self, a, b):
        """
        :type a: str
        :type b: str
        :rtype: str
        """
        la = len(a)
        lb = len(b)
        c = [None for i in range(max(la,lb))]
        if la > lb:            # 把两个扩充到统一长度，方便逐位相加
            b = '0' * (la-lb) + b
        elif la < lb:
            a = '0' * (lb-la) + a

        sign = 0
        for i in range(max(la, lb)-1, -1, -1):
            if int(a[i]) + int(b[i]) + sign == 2:
                c[i] = 0
                sign = 1
            elif int(a[i]) + int(b[i]) + sign == 3:
                c[i] = 1
                sign = 1
            else:
                c[i] = int(a[i]) + int(b[i]) + sign
                sign = 0
        c = ''.join(map(str, c))
        if sign == 1:
            c = '1' + c
        return c

解法二：位运算

考虑a='101', b='111'，转为二进制数后，则a^b = 010，a & b =101，期望的输出为'1000'

分析：

• a & b要产生为1的位，那么a和b对应位就都是1，也就会产生进位。所以a & b =101意味着第一位和第三位都会产生进位，所以进位结果为a & b << 1，即1010。
• a^b要产生为1的位，那么a和b对应位必有一个0一个1，此时异或相当于求和（对这一位来说）。
• a^b再与a&b<<1异或，相当于加上进位，两者逐位相与，又得到新的进位表示。重复这一过程，直到进位为0。

class Solution(object):
    def addBinary(self, a, b):
        """
        :type a: str
        :type b: str
        :rtype: str
        """
        a, b = int(a, 2), int(b, 2)
        while True:
            a, b = a ^ b, (a & b) << 1
            if b == 0:
                break
        return bin(a)[2:]

 

参考：

https://leetcode-cn.com/problems/add-binary/solution/er-jin-zhi-qiu-he-by-leetcode/