[프로그래머스] Level1 - 나머지가 1이 되는 수 찾기 (Python)

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

---

제한사항

• 3 ≤ n ≤ 1,000,000

---

입출력 예

n	result
10	3
12	11

입출력 예 설명

입출력 예 #1

• 10을 3으로 나눈 나머지가 1이고, 3보다 작은 자연수 중에서 문제의 조건을 만족하는 수가 없으므로, 3을 return 해야 합니다.

입출력 예 #2

• 12를 11로 나눈 나머지가 1이고, 11보다 작은 자연수 중에서 문제의 조건을 만족하는 수가 없으므로, 11을 return 해야 합니다.

내 문제 풀이

• 주어진 숫자 n에 대해서 1부터 반복문을 시작해서 나머지가 1인 경우를 찾으면 바로 해당 숫자를 return
• 나머지가 1이 아닌 경우는 나누려는 숫자를 1 증가 시킴

def solution(n):
    cnt = 1
    
    while True:
        if n % cnt == 1:
            return cnt
        else:
            cnt += 1
    

아쉬운 점

• 위 문제도 1문제로 풀 수 있다!

return min([i for i in range(1,n) if n % i == 1])

list comprehension을 이용해서 list 값 중에서 제일 작은 값을 return 해주면 된다....