1. 程式人生 > >Leetcode:67. 二進位制求和

Leetcode:67. 二進位制求和

給定兩個二進位制字串,返回他們的和(用二進位制表示)。

輸入為非空字串且只包含數字 1 和 0

示例 1:

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

示例 2:

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

解題思路:

和字串相加的演算法一致,考慮進位,只不過對2求餘。如果連這個都不會,。。。,那就看下面程式碼。

C++
class Solution {
public:
    string addBinary(string a, string b) {
        while (int(a.size()) >= 2 && a[0] == '0') a.erase(a.begin());
        while (int(b.size()) >= 2 && b[0] == '0') b.erase(b.begin());
        if (a == "0") return b;
        if (b == "0") return a;
        if (a.size() < b.size()) swap(a, b);
        b = string(int(a.size() - b.size()), '0') + b;//對齊
        int sgn = 0, data;
        for (int i = int(a.size()); i >= 1; i--) {
            data = int(a[i - 1] - '0') + int(b[i - 1] - '0') + sgn;
            sgn = data / 2;
            data = data % 2;
            a[i - 1] = data + '0';
        }
        if (sgn == 1) a = string(1, '1') + a;
        return a;
    }
};