swift언어 reduce 고차함수 예제, 백준 1037 약수 풀이

 

 

 

 

---

백준 1037번, 약수 문제설명

오늘은 백준의 1037번 문제, 약수를 풀어보겠습니다. 오늘 문제풀이는 문제의 풀이과정이 아닌, reduce를 이런식으로 사용할 수 있다는 것에 초점을 두고 진행하겠습니다. 

 

 

 

 

시간 제한은 2초, 메모리 제한은 512MB입니다. 진짜 약수가 주어졌을때 진짜 약수들을 갖는 수를 구하는 문제입니다.

 

 

 

 

먼저, 진짜약수의 갯수를 첫줄에 입력받고, 그 갯수만큼 진짜갯수의 약수를 입력받습니다. 위의 경우, 진짜약수 4, 2를 갖는 수를 구하는 문제로 정답은 8입니다. (진짜약수에 1은 포함하지 않습니다.)

 

결론부터 말슴드리자면, 해당 문제는 주어지는 진짜약수의 최댓값, 최솟값을 활용해서 풀 수 있습니다.
- 진짜약수의 갯수가 1개인 경우 : (진짜약수 최댓값) * (진짜약수 최댓값)

- 진짜약수의 갯수가 2개 이상인 경우 : (진짜약수 최댓값) * (진짜약수 최솟값) 

이 됩니다. 본문제는 O(N)의 시간복잡도로 풀 수 있으며, 진짜약수의 갯순는 최대 50개에 불과하므로 입력최적화를 꼭 사용할 필요는 없는 기초문제입니다. 또한 이번 포스팅에서 사용할 reduce(into:)를 굳이 사용할 필요 없습니다.

 

하지만 글 제목에서도 작성했듯이, swift언어의 reduce(into:)를 사용해서 본 문제 풀어보겠습니다. 

 

 

 

---

백준 1037번, 약수 swift 문제풀이

먼저 String 입력을 받아서 [Int]타입 배열로 반환해주는 클로져를 미리 선언했습니다. 본 문제는 입력받는 양이 크지 않기 때문에 굳이 위와 같이 입력 받을 필요없이 Int(readLine()!)!이나 split함수를 활용해서 처리해도 문제 없습니다. 

 

 

 

 

진짜갯수의 약수, 진짜갯수 리스트 입력을 받아줍니다. 이제 이어서 진짜갯수의 최대 / 최솟값을 구해주면 됩니다. 배열의 최댓값 / 최솟값을 구하는 방법은 배열의 max()!, min()! 메서드를 사용하거나, for / forEach문으로 순회하거나 등등... 방법은 다양하니 다양한 방법으로, 입맛대로 접근하시면 됩니다. 

 

저의 경우에는 reduce(into:) 함수를 사용해보도록 하겠습니다. 

 

 

 

 

pair는 (최솟값, 최댓값)을 갖는 (Int, Int) 튜플 타입의 값을 저장합니다. 이게 곧 진짜 약수의 최솟값, 최댓값을 가진 상수가 됩니다. 

reduce는 초기값을 기준으로 배열의 값을 참조해서 특정 연산을 거듭, 하나의 완성값을 만들고 싶을 때 사용할 수 있습니다.

 

22 ~ 23행) 현재 거치는 진짜약수의 값이 최솟값이면 갱신해주고, 최댓값일때에도 갱신해주는 코드입니다. 이렇게 $0.0, $0.1에 저장된 최종 값은 pair 상수에 저장이 됩니다. 

$0는 현재 누적되고 있는 튜플 값이며 타입은 (Int, Int)입니다. $1은 현재 순회하고 있는 arr배열의 값입니다.

 

 

 

 

앞서 말씀드렸듯이, N이 1, 즉 진짜 약수가 1개 뿐이라면 최댓값의 제곱을, 2개 이상이라면 최댓값 * 최솟값을 출력해주면 진짜약수를 가진 수를 출력할 수 있습니다.