LeetCode알고리즘 Add Binary

오늘은 67번 문제인 Add  Binary를 풀어보겠습니다.
쉬운 난이도며 48%의 성공률을 보입니다.

이 문제의 주제는 비트입니다.

먼저 문제를 살펴보겠습니다. 

문자열 a와 b가 주어지고 그것들을 합을 이진 문자열로 반환해주면 됩니다.

주어진 예제를 살펴봅시다.

아래는 1번 예제와 2번 예제를 푼 것입니다.

이 문제를 풀기 위해서 문자열로 받는 a와 b를 인덱스 접근에 가능하도록 배열로 만들어주었습니다.
그리고 answer이라는 곳에 최종 정답이 담 길거고 carry는 캐리 값을 나타내고 i와 j는 각 문자열의 길이-1을 나타냅니다.

func addBinary(_ a: String, _ b: String) -> String {
    let a = Array(a) 
    let b = Array(b) 
    var answer = [""]
    var carry = 0
    var i = a.count - 1 
    var j = b.count - 1 

    while i >= 0 || j >= 0 || carry > 0 {
        var sum = carry

        if i >= 0 {
            sum = sum + Int(String(a[i]))!
            i = i - 1
        }
        
        if j >= 0 {
            sum = sum + Int(String(b[j]))!
            j = j - 1
        }
        answer.append("\(sum % 2)")
        carry = sum / 2
    }

    return answer.reversed().joined()
}

while루프에 들어오는 조건은 i가 0보다 크거나 같고, j가 0보다 크거나 같고, carry가 0보다 크다는 조건입니다.
먼저 sum에 carry를 넣어주고 a와 b를 끝에서부터 더해줍니다.
각각의 끝자리를 더하고나면 i, j를 -1해 주어 앞자리를 더할 수 있도록해줍니다.
answer에 sum % 2를 계산해서 붙여주고 carry에는 sum / 2를 계산해서 넣어줍니다.

이렇게 했을 때 결과는 아래와 같이 나오네요. 

근데 answer을 문자열 배열로 초기화해서 배열에 reversed(), joined()를 쓰는 게 아닌
아래와 같이 문자열로 초기화해서 string interporation을 사용하면
코드가 조금 더 깔끔해지고 필요 없는 연산을 줄일 수 있을 것 같네요.

func addBinary(_ a: String, _ b: String) -> String {
    let a = Array(a)
    let b = Array(b)
    var answer = ""
    var carry = 0
    var i = a.count - 1
    var j = b.count - 1

    while i >= 0 || j >= 0 || carry > 0 {
        var sum = carry

        if i >= 0 {
            sum = sum + Int(String(a[i]))!
            i = i - 1
        }
        
        if j >= 0 {
            sum = sum + Int(String(b[j]))!
            j = j - 1
        }

        answer = "\(sum % 2)" + answer
        carry = sum / 2
    }

    return answer
}

이렇게 했을 때 결과는 아래와 같이 나옵니다. 

으으 비트는 항상 어렵네요.

resource:
https://leetcode.com/problems/add-binary/