题目
给定两个二进制字符串,返回他们的和(用二进制表示)。
输入为非空字符串且只包含数字 1 和 0。
示例 1:
输入: a = "11", b = "1"
输出: "100"
示例 2:
输入: a = "1010", b = "1011"
输出: "10101"
代码模板:
class Solution {
public String addBinary(String a, String b) {
}
}
分析
这道题和两个有序数组的合并很类似,只是多加了一个判断。这里我们设置一个进位carry,每次取两个数最后一位相加再加上进位作为一个num,再根据这个num来判断下一个的进位和当前值是多少。这样一直循环下去。最后注意如果进位不为0的话,就append一个1,最后把这个result反转一下就能得到我们要的结果了。
解答
class Solution {
public String addBinary(String a, String b) {
int aindex = a.length() - 1;
int bindex = b.length() - 1;
StringBuffer result = new StringBuffer();
//进位
int carry = 0;
while (aindex >= 0 || bindex >= 0){
int sum = carry;
if(aindex>=0){
sum = sum + a.charAt(aindex)-'0';
aindex --;
}
if(bindex>=0){
sum = sum + b.charAt(bindex)-'0';
bindex --;
}
carry = addString(result,sum);
}
while (aindex >= 0){
int sum = (a.charAt(aindex)-'0')+carry;
aindex --;
carry = addString(result,sum);
}
while(bindex >= 0){
int sum = (b.charAt(bindex)-'0')+carry;
bindex --;
carry = addString(result,sum);
}
if(carry != 0){
result.append(1);
}
return result.reverse().toString();
}
public int addString(StringBuffer result,int sum){
if(sum == 0){
result.append(0);
return 0;
}else if(sum == 1){
result.append(1);
return 0;
}else if(sum == 2){
result.append(0);
return 1;
}else if (sum ==3){
result.append(1);
return 1;
}
return 0;
}
}