[코테] 나머지가 1인 수 구하기

문제

나머지가 1인 수 구하기

문제설명

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

해결책

자연수 n이 나머지가 1이 나올 수 있는 경우가 무엇이 있을까? 나는 그냥 n보다 작은 수를 다 나눈 후 나머지 1이 나오는 수들을 리스트에 담아준 후 에 가장 먼저 나오는 수를 반환했다.

코드

나의코드

def solution(n):
    answer = 0
    ans = []
    for x in range(2, n):
        if(n%x == 1):
            ans.append(i)
    
    answer = ans[0]
    return answer

타인의 코드

def solution(n):
    return [x for x in range(1,n+1) if n%x==1][0]

소감

한줄로 줄인 사람의 코드를 보면서 아직도 파이썬의 리스트 list comprehesion 을 제대로 사용하지 못하고 있다는 것을 다시 한번 정리하고 넘어가야겠다.