💯 Daily LeetCode Challenge Day_06 - Plus One

Account Login - LeetCode

---

문제

Given a non-empty array of digits representing a non-negative integer, plus one to the integer.

The digits are stored such that the most significant digit is at the head of the list, and each element in the array contain a single digit.

You may assume the integer does not contain any leading zero, except the number 0 itself.

Example 1:

Input: [1,2,3]
Output: [1,2,4]
Explanation: The array represents the integer 123.

Example 2:

Input: [4,3,2,1]
Output: [4,3,2,2]
Explanation: The array represents the integer 4321.

---

이제는 이렇게 Int Array 나 String으로 주면서 덧셈을 하라는 문제들은

그냥 Int 자료형의 MAX 값 보다 덧셈값이 커져서 오버플로우 일어나기 떄문에 digit 별로 처리해야 되는 문제로 

바로 생각이 든다.

carry 변수를 사용해 맨처음 1을 더하고 carry 유무에 따라 값을 바꿔준다.

class Solution {
    func plusOne(_ digits: [Int]) -> [Int] {
        var arr = digits
        var answer : [Int] = []
        var carry = 0
        for i in (0..<digits.count).reversed() {
            var digit = arr[i] + carry
            if i == digits.count - 1 { digit += 1 }
            // print(digit, terminator: " " )
            answer.append(digit%10)
            carry = digit/10
            
        }
        if carry != 0 {
            answer.append(carry)
        }
        return answer.reversed()
    }
}

---

혹시나 싶어 Int 로 바꾸어 +1 만 더한 뒤 다시 Int array 로 만들어 리턴해 보았다.

class Solution {
    func plusOne(_ digits: [Int]) -> [Int] {
        var number = digits.map{String($0)}.joined(separator: "")
        var tmp = Int(number)! + 1
        let answer = String(tmp).compactMap {$0.wholeNumberValue}
        return answer
    }
}

 

응안되 돌아가 

 

비슷한 방식의 덧셈 코드이다.

Swift 배열 처리