[프로그래머스] 나머지가 1이 되는 수 찾기 Kotlin

안녕하세요. 이번에는 프로그래머스 나머지가 1이 되는 수 찾기를 풀어보려고 합니다.

 

https://school.programmers.co.kr/learn/courses/30/lessons/87389

프로그래머스

 

---

Problem

자연수 n이 매개변수로 주어집니다. n을 x로 나눈 나머지가 1이 되도록 하는 가장 작은 자연수 x를 return 하도록 solution 함수를 완성해 주세요. 답이 항상 존재함은 증명될 수 있습니다.

 

[제한 사항] : ○ 3 ≤ n ≤ 1,000,000 

사진 1. 입출력 예

---

Solution

class Solution {
    fun solution(n: Int): Int {
        var answer: Int = 0
        
        for(i in 2..n) {
            if(n % i == 1) {
                answer = i
                break;
            }
        }
        
        return answer
    }
}

 i의 값을 n까지 1씩 증가시키면서 나머지가 1이 나오는 i값을 리턴시키는 방식으로 풀어보았다. 문제를 푼 후 조금 더 공부하기 위해 다른 사람이 작성한 코드를 보게 되었는데 first 함수를 사용해서 문제를 해결한 것을 확인할 수 있었다. 여기서 first 함수를 알게 되었다.

 

1. First 함수를 사용한 풀이

class Solution {
    fun solution(n: Int): Int {
        var answer: Int = 0
        
       	answer = (1..n).first {n % it == 1 }
        
        return answer
    }
}

---

Performance

1. 내가 작성한 코드

사진 2. 실행 결과

2. first 함수를 사용한 코드

사진 3. first함수 실행 결과

 

사용한 코드들의 속도를 비교해보면 first함수보다 for문 풀이법이 더 빠른 것을 확인 할 수 있었다. 하지만 first함수를 사용하면 코드의 가독성이 증가하는 것을 알 수 있었다. 제한 사항에 편한 방법으로 사용하면 될 것 같다.