67. 二进制求和
给定两个二进制字符串,返回他们的和(用二进制表示)。
输入为非空字符串且只包含数字 1
和 0
。
示例 1:
输入: a = "11", b = "1"
输出: "100"
示例 2:
输入: a = "1010", b = "1011"
输出: "10101"
思路:
为了遍历起来方便,我先把两个字符串对齐,将短的前面补0,不影响结果。
设置进位数count,初始为0;从最后一个字符开始向前循环,两个字符串的第i个元素相加并加上count,得到该位置上的“和”,叫做result,它有四种取值,分别为 0,1,2,3。
循环完之后若 count 为 1 ,说明还要进位,则在结果字符串最前面补上 1 。
/**
* @param {string} a
* @param {string} b
* @return {string}
*/
var addBinary = function(a, b) {
var al = a.length, bl = b.length;
var maxl;
if (al > bl) {
maxl = al;
b = Array(al - bl + 1).join('0') + b;
}
if (al < bl) {
maxl = bl;
a = Array(bl - al + 1).join('0') + a;
}
var count = 0;
var res = '';
var i;
for (i = a.length-1; i >= 0; i--) {
switch(parseInt(a[i]) + parseInt(b[i]) + count) {
case 0:
res = '0' + res;
count = 0;
break;
case 1:
res = '1' + res;
count = 0;
break;
case 2:
res = '0' + res;
count = 1;
break;
case 3:
res = '1' + res;
count = 1;
break;
}
}
if (count === 1) {
res = '1' + res;
}
return res;
};