[swift] 백준 - 1920번: 수 찾기

1920번: 수 찾기

문제 푸는 데 있어 도움이 되도록 나의 풀이와 개선된 풀이를 올립니다.
또한 풀이 후 다른 사람의 풀이를 보고 참고할만한 풀이도 올립니다.
- 문제에 따라 나의 풀이만 있을 수 있습니다.
- 해당 풀이들은 풀이 중 하나일 뿐 최선의 풀이는 아닐 수 있습니다.

 

---

 

문제 설명

 

- 시간제한 안에 해당하는문자(숫자)가 배열에 포함하는지 탐색하라.

 

---

 

나의 풀이

import Foundation

func binarySearch(_ array: [Int], _ value: Int) -> Int? {
    guard !array.isEmpty else { return nil }
    
    var left = 0
    var right = array.count - 1
    
    while left <= right {
        let middleIndex = (left + right) / 2
        let middleValue = array[middleIndex]
        
        if middleValue > value {
            right = middleIndex - 1
        }
        if middleValue < value {
            left = middleIndex + 1
        }
        if middleValue == value {
            return middleIndex
        }
    }
    return nil
}

let N = Int(readLine()!)!
let A = readLine()!.split(separator: " ").map { Int($0)! }.sorted()
let M = Int(readLine()!)!
let B = readLine()!.split(separator: " ").map { Int($0)! }

for i in B { binarySearch(A, i) == nil ? print(0) : print(1) }

 

- 이진탐색을 통해 시간제한을 통과할 수 있었습니다.

 

-약 200~250ms

 

---

 

 

참고할만한 풀이

import Foundation

let N = Int(readLine()!)!
let A = Set(readLine()!.split(separator: " ").map { Int(String($0))! })
let M = Int(readLine()!)!
let B = readLine()!.split(separator: " ").map { Int(String($0))! }
var result = ""

for i in B { result += A.contains(i) ? "1\n" : "0\n" }
print(result)

 

- 중첩되는 숫자가 많은지 Set을 통해

시간제한 안에 풀 수 있습니다.

 

- 나의 풀이보다 간결해졌습니다.

 

- 삼항 연산자로 결과를 바로 출력하는거 보다

결과를 출력할 빈 문자열을 하나 만들어 놓고

print할 문자들을 추가하여

한번에 출력하니 시간을 단축시킬 수 있었습니다.

 

- 약 150ms

 

---