题目:
给定两个二进制字符串,返回他们的和(用二进制表示)。
输入为非空字符串且只包含数字 1 和 0。
示例 1:
输入: a = "11", b = "1"
输出: "100"
示例 2:
输入: a = "1010", b = "1011"
输出: "10101"
思路:
从后往前计算,记录进位,直到首位,判断是否还需要再进位。
代码实现:
class Solution {
public:
string addBinary(string a, string b) {
int i=a.length()-1;
int j=b.length()-1;
int m=0,n=0,k=0,sum=0;//k记录进位,sum记录各位数的和
string s = "",temp="";
while(i>=0||j>=0){
if(i>=0){
m = a[i]-'0';
}else{
m = 0;
}
if(j>=0){
n = b[j]-'0';
}else{
n = 0;
}
sum = m+n+k;
temp = to_string(sum%2);
if(sum>=2){
s = s+ temp ;
k = 1;
}else {
s = s+ temp ;
k = 0;
}
i--;j--;
}
if(sum>=2){
s+="1";
}
reverse(s.begin(),s.end());
return s;
}
};
效率:(有点低)
Java实现:
class Solution {
public String addBinary(String a, String b) {
StringBuilder ans = new StringBuilder();
int ca = 0;
for(int i = a.length() - 1, j = b.length() - 1;i >= 0 || j >= 0; i--, j--) {
int sum = ca;
sum += i >= 0 ? a.charAt(i) - '0' : 0;
sum += j >= 0 ? b.charAt(j) - '0' : 0;
ans.append(sum % 2);
ca = sum / 2;
}
ans.append(ca == 1 ? ca : "");
return ans.reverse().toString();
}
}
效率: