[Swift] 알고리즘 공부 - 나머지가 1이 되는 수 찾기

문제 설명

 

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

 

제한사항

• 3 ≤ n ≤ 1,000,000

 

입출력 예

 

n                             result

10	3
12	11

 

설명

 

1. 1부터 n/2까지 (n/2를 넘어가면 나머지가 1이 되는 수는 n-1밖에 없다) 한번씩 n을 나눠본다.

2. 만약 나머지가 1이 되는 수가 있다면 그수를 return 하고 없다면 n-1를 return

 

import Foundation

func solution(_ n:Int) -> Int {

    for i in 1...n/2 {
        if n % i == 1 { return i }
    }
    
    return n-1
}