[Swift] 프로그래머스 연습 문제(Level 1) : 소수 찾기
실제 코딩테스트의 문제를 통하여 알고리즘 분석과 코딩 능력을 향상시킵니다.
#. 구독 대상
- 기본 알고리즘을 코딩 테스트 문제를 통하여 학습하고 싶으신 분
- 취업 및 이직을 준비하고 계신 개발자
- Swift를 통하여 코딩 테스트 문제를 살펴보고 이해를 하고 싶으신 분
- 코딩 테스트에 대한 거부감을 없애기 위하여 기초부터 하나씩 공부해보고 싶으신 분
- 기타 알고리즘과 문제 해결 능력에 대해서 관심이 있는 모든 개발자분
참고 사항
본 코딩 테스트 문제에 대한 설명 및 해결 방안은 최적의 답이 아닐 수 있습니다.
본 강의에서 지향하는 목표는 바로 특정 문제에 대한 최적의 해결 방법을 찾기보다는 특정한 문제에 대해서 충분히 이해할 수 있고 다양한 방법을 통하여 해결하는 방법을 찾고 향상시키는데 그 목적이 있습니다.
좀 더 좋은 알고리즘 및 코드가 있으시다면 언제든지 본 게시물의 댓글을 통해서 제시해주시면 감사하겠습니다.
코딩 테스트 문제
먼저 오늘 살펴볼 문제에 대해서 먼저 살펴보겠습니다.
programmers.co.kr/learn/courses/30/lessons/12921
코딩테스트 연습 - 소수 찾기
1부터 입력받은 숫자 n 사이에 있는 소수의 개수를 반환하는 함수, solution을 만들어 보세요. 소수는 1과 자기 자신으로만 나누어지는 수를 의미합니다. (1은 소수가 아닙니다.) 제한 조건 n은 2이상
programmers.co.kr
문제 설명
1부터 입력받은 숫자 n 사이에 있는 소수의 개수를 반환하는 함수, solution을 만들어 보세요.
소수는 1과 자기 자신으로만 나누어지는 수를 의미합니다.
(1은 소수가 아닙니다.)
제한 조건
- n은 2이상 1000000이하의 자연수입니다.
입출력 예
n | result |
10 | 4 |
5 | 3 |
문제 분석
오늘 문제는 숫자를 입력 받고 그 숫자 사이에 있는 소수의 갯수를 반환하는 문제입니다.
기본적인 소수에 대한 정의를 이해하고 특정한 수의 소수 여부를 판단하면 문제의 답을 찾을 수 있는 문제입니다.
알고리즘
위의 내용을 기반으로 하여 본 문제를 해결하기 위한 알고리즘을 살펴보면 아래와 같습니다.
- 소수란 자신보다 작은 수들 중에서 1과 자기 자신을 제외하고는 그 어떠한 수로도 나누어지지 않는 수를 뜻합니다.
- 따라서 특정 숫자에 대한 소수 여부를 판별할 수 있는 함수를 우선 생성합니다.
- 1을 제외하고 2부터 입력받은 수 사이의 소수를 찾기위하여 완전 탐색(Brute Force) 방식(순차 탐색)으로 하나씩 비교하면서 소수의 갯수를 체크합니다.
코드 설명
그렇다면 위의 알고리즘에 대해서 하나씩 살펴보면서 코드로 작성을 해보도록 하겠습니다.
- 소수란 자신보다 작은 수들 중에서 1과 자기 자신을 제외하고는 그 어떠한 수로도 나누어지지 않는 수를 뜻합니다.
- 따라서 특정 숫자에 대한 소수 여부를 판별할 수 있는 함수를 우선 생성합니다.
func isPrime(number: Int) -> Bool {
var result: Bool = true
if number < 2 {
result = false
} else if number > 2 {
for i in 2 ..< number {
if number % i == 0 {
result = false
}
}
}
return result
}
- 1을 제외하고 2부터 입력받은 수 사이의 소수를 찾기위하여 완전 탐색(Brute Force) 방식(순차 탐색)으로 하나씩 비교하면서 소수의 갯수를 체크합니다.
var result = 0
for i in 2...n {
if (isPrime(number: i)) {
result += 1
}
}
위의 코드들을 모두 조합하여 최종 코드를 완성하면 아래와 같습니다.
최종 코드
func solution(_ n:Int) -> Int {
var result = 0
for i in 2...n {
if (isPrime(number: i)) {
result += 1
}
}
return result
}
func isPrime(number: Int) -> Bool {
var result: Bool = true
if number < 2 {
result = false
} else if number > 2 {
for i in 2 ..< number {
if number % i == 0 {
result = false
}
}
}
return result
}
이상으로 오늘 제시한 문제에 대해서 분석 및 코드를 작성해 보았습니다.
그럼 주말 저녁 잘 마무리하시길 바랍니다.
감사합니다.
[참고 자료(References)]
[1] 프로그래머스 - 소수 찾기 : programmers.co.kr/learn/courses/30/lessons/12921