题目
leetcode67. 二进制求和
给你两个二进制字符串,返回它们的和(用二进制表示)。
输入为 非空 字符串且只包含数字 1 和 0。
示例 1:
输入: a = “11”, b = “1”
输出: “100”
示例 2:
输入: a = “1010”, b = “1011”
输出: “10101”
提示:
每个字符串仅由字符 ‘0’ 或 ‘1’ 组成。
1 <= a.length, b.length <= 10^4
字符串如果不是 “0” ,就都不含前导零。
代码
- 跟 Code17 加一 大致思路一致,在考虑进位时稍复杂些,需要从字符转为数字,且进制是二进制而不是十进制。
- C代码
- 思路:
- C代码涉及 malloc , 所以需要先确认申请资源的大小,但一开始并不能确定是否有最后的进位,所以先申请字符串a,b中较长长度大小等同的资源,
- 最后若有进位,重新申请一块资源,将结果拷贝过去,并释放第一块资源。
- 最后若无进位,直接返回结果。
- C代码涉及 malloc , 所以需要先确认申请资源的大小,但一开始并不能确定是否有最后的进位,所以先申请字符串a,b中较长长度大小等同的资源,
#include <string.h>
#include <malloc.h>
char* addBinary(char* a, char* b) {
int a_len = strlen(a);
int b_len = strlen(b);
int len = a_len > b_len ? a_len : b_len;
// 申请第一块资源
char* s = (char*)malloc(sizeof(char) * (len + 1));
s[len] = '\0';
int carry = 0; // 进位
int a_idx = a_len - 1; // 最后一个字符下标向前遍历
int b_idx = b_len - 1; // 最后一个字符下标向前遍历
int s_idx = len - 1; // 用于结果填充下标
while(a_idx >= 0 || b_idx >= 0){
int aidx_num = a_idx < 0 ? 0 : a[a_idx] - '0';
int bidx_num = b_idx < 0 ? 0 : b[b_idx] - '0';
s[s_idx--] = (aidx_num + bidx_num + carry) % 2 + '0'; // 加法逻辑:本位用模
carry = (aidx_num + bidx_num + carry) / 2; // 加法逻辑:进位用除
--a_idx;
--b_idx;
}
if (carry) {
// 有最后进位
// 重新申请资源,比第一块资源长度 + 1, 用于放最后进位
char* res = (char*)malloc(sizeof(char) * (len + 1 + 1));
res[0] = carry + '0'; // 第1位放最后进位
memcpy(res + 1, s, len + 1); // 拷贝剩余结果
free(s); // 释放第一块资源
return res;
}else{
return s; // 无最后进位,直接返回
}
}
- C++代码
- 不涉及资源申请问题,在string中采用前插即可。
#include <string>
using namespace std;
class Solution {
public:
string addBinary(string a, string b) {
string res;
int carry = 0;
auto a_it = a.rbegin();
auto b_it = b.rbegin();
while(a_it < a.rend() || b_it < b.rend()){
int a_num = a_it >= a.rend() ? 0 : *a_it - '0';
int b_num = b_it >= b.rend() ? 0 : *b_it - '0';
res.insert(res.begin(), (a_num + b_num + carry) % 2 + '0');
carry = (a_num + b_num + carry) / 2;
++a_it;
++b_it;
}
if (carry){
res.insert(res.begin(), carry + '0');
}
return res;
}
};
测试
#include <iostream>
int main() {
{
char* a = "1010";
char* b = "1011";
char* s = addBinary(a, b);
cout << s << endl;
free(s);
}
{
string a = "1010";
string b = "1011";
Solution s;
cout << s.addBinary(a, b);
}
cin.get();
return 0;
}
- 结果
10101
10101