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

나머지가 1이 되는 수 찾기

문제 설명

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

제한 조건

• 3 ≤ n ≤ 1,000,000

풀이

각 w, h 를 정렬하면 무조건 작은 값이 왼쪽에 오게 된다. 그리고 각 max(w) * max(h)를 곱하면 된다.

def solution(n):
    for i in range(2, n // 2):
        if n % i == 1:
            return i
    return n - 1