力扣(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;
    }
};