[Swift] n의 배수 고르기 문제 회고

안녕하세요! 피피아노입니다 🎵

 

이번 포스팅에서는 n의 배수 고르기 알고리즘 문제 회고를 작성해보려고 합니다.

 

그럼 바로 시작하겠습니다!

 

문제

n의 배수 고르기 문제는 정수 n과 정수 배열 numlist가 매개변수로 주어지고, numlist에서 n의 배수가 아닌 수들을 제거한 배열을 return하도록 코드를 작성하는 문제입니다.

 

내가 풀었던 방식

제가 풀었던 문제 방식은

import Foundation

func solution(_ n:Int, _ numlist:[Int]) -> [Int] {
    var result: [Int] = []
    
    for i in numlist {
        if i % n == 0 {
            result.append(i)
        }
    }
    return result
}

이렇게 작성하였습니다. 

 

먼저 result 변수를 선언해주고 for문을 사용해서 numlist 안에 있는 정수들을 모두 한번씩 돌도록 작성하였고 for문 안에서 if문에서 n의 배수에 해당되는 수들만 append 메서드로 result 변수에 넣어주었습니다.

 

*append 메서드란?

: append 메서드는 Swift에서 배열에 새로운 요소를 추가할 때 사용하는 메서드로 배열의 마지막에 새로운 값을 넣을 수 있게 해줍니다.

 

물론 이렇게 해도 의도한대로 코드는 돌아가지만 테스트 결과 2군데에서 시간초과가 발생하는 것을 파악 했습니다.

 

원인을 살펴보았고 반복문을 통해 매번 numlist 요소를 확인하고 append하는 과정에서 효율성이 떨어질 수도 있겠다라는 생각이 들었습니다. 

해결 코드

이를 해결하기 위해 다른 메서드를 사용하는 것을 고려해보았고 좀 더 빠르게 원하는 조건만 뽑아 내기 위해서 filter 메서드를 사용해보았습니다.

import Foundation

func solution(_ n:Int, _ numlist:[Int]) -> [Int] {
    return numlist.filter{ $0 % n == 0 }
}

그렇게 해서 작성한 코드가 이 코드입니다. 한 눈에 봐도 훨씬 간결하죠?

 

이렇게 하고 실행해보니 시간복잡도를 해결할 수 있었습니다.

 

정리

정리를 해보자면 처음에 저는 for문과 if문을 이용해서 원하는 조건만 특정 변수에 넣어주고 해당 변수를 리턴하는 방식으로 코드를 구현했지만 시간 복잡도에서 문제가 발생하였고 이러한 부분을 해결하기 위해 filter 메서드를 이용한 코드를 작성하였습니다.

 

느낀 점

항상 코드를 구현 하는 것에만 급급해서 자꾸 시간 복잡도에서 문제가 생기는데 조금 더 여유 있게 생각해보고 구현하는 것에 성공 했다면 더 간결하게 작성하거나 다른 방법으로 구현하는 방법도 생각을 해봐야겠습니다.

---

감사합니다.

 

잘못된 내용이 있거나 더 좋은 내용 피드백은 언제나 환영합니다!

궁금하신 부분은 댓글로 질문 부탁드립니다!