Leetcode67. 二进制求和
题目:
给定两个二进制字符串,返回他们的和(用二进制表示)。
输入为非空字符串且只包含数字 1 和 0。
示例 1:
输入: a = “11”, b = “1”
输出: “100”
题解:
整体思路是将两个字符串较短的用 0补齐,使得两个字符串长度一致,然后从末尾进行遍历计算,得到最终结果。
本题解中大致思路与上述一致,但由于字符串操作原因,不确定最后的结果是否会多出一位进位,所以会有 2 种处理方式:
第一种,在进行计算时直接拼接字符串,会得到一个反向字符,需要最后再进行翻转
第二种,按照位置给结果字符赋值,最后如果有进位,则在前方进行字符串拼接添加进位
scala代码如下:
def addBinary(a: String, b: String): String = {
val aLength = a.length
val bLength = b.length
val maxLen = Math.max(aLength, bLength)
// 从个位开始加,个位在字符串的右边
// 代码访问从左到右,因此先反转一下
val sbA = new StringBuilder(a).reverse
val sbB = new StringBuilder(b).reverse
while (sbA.length < maxLen) {
sbA.append("0")
}
while (sbB.length < maxLen) {
sbB.append("0")
}
val builder = new StringBuilder()
// 进位,初始时进位为 0
var carry = 0
var num1 = 0
var num2 = 0
for (i <- 0 until maxLen) {
// 当前字符的 ASCII 值减去 '0' 的 ASCII 值,相当于将这个字符转换成数值
num1 = sbA.charAt(i) - '0'
num2 = sbB.charAt(i) - '0'
if (carry + num1 + num2 > 1) {
// 因为是二进制,所以多余 2 的部分要减去
builder.append(carry + num1 + num2 - 2)
// 表示要进位
carry = 1
} else {
builder.append(carry + num1 + num2)
carry = 0
}
}
// 对于最高位还要进位的情况,需要单独判断
if (carry == 1) {
builder.append("1")
}
// 最后不要忘记再反转一次
builder.reverse.toString()
}