力扣(C++):二進制求和

題目描述

給定兩個二進制字符串，返回他們的和（用二進制表示）。
輸入爲非空字符串且只包含數字 1 和 0。

示例 1:

輸入: a = “11”, b = “1”
輸出: “100”

示例 2:

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

來源：力扣（LeetCode）
鏈接：二進制求和

代碼示例:

class Solution {
public:
    string addBinary(string a, string b) {
        int lenA = a.size();
        int lenB = b.size();
        //先讓兩個字符串等長
        while (lenA < lenB)
        {
            a = '0' + a;
            ++lenA;
        }
        while (lenA > lenB)
        {
            b = '0' + b;
            ++lenB;
        }
        //滿2進1
        for (int i = lenA - 1; i > 0; --i)
        {
            a[i] = a[i] - '0' + b[i];
            if (a[i] >= '2')
            {
                a[i] = (a[i] - '0') % 2 + '0';
                a[i - 1] = a[i - 1] + 1;
            }
        }
        //判斷首位
        a[0] = a[0] - '0' + b[0];
        if (a[0] >= '2')
        {
            a[0] = (a[0] - '0') % 2 + '0';
            a = '1' + a;
        }
        return a;
    }
};