LeetCode/67_AddBinary.py at master · brettchien/LeetCode · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
class Solution:
    # @param {string} a
    # @param {string} b
    # @return {string}
    def addBinary(self, a, b):
        result = ""
        if len(b) > len(a):
            a, b = b, a
        carry = False
        while len(a):
            x = a[-1]
            a = a[:-1]
            y = '0'
            if len(b):
                y = b[-1]
                b = b[:-1]
            digit = '0'
            if carry:
                if x == '1' and y == '1':
                    digit = '1'
                elif x == '0' and y == '0':
                    digit = '1'
                    carry = False
                else:
                    digit = '0'
            else:
                if x == '1' and y == '1':
                    digit = '0'
                    carry = True
                elif x == '0' and y == '0':
                    digit = '0'
                else:
                    digit = '1'
            result = digit + result
        if carry:
            result = "1" + result
        return result


if __name__ == "__main__":
    sol = Solution()
#    print sol.addBinary("1010", "1011")
    print sol.addBinary("11", "1")