Python/Coding Test

[Coding Test] 나머지가 1이 되는 수 찾기

gangee 2023. 2. 8. 22:25
728x90
반응형

프로그래머스 코딩테스트 1단계 : 나머지가 1이 되는 수 찾기

문제 설명

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

문제 풀이

  • 반복문의 범위 x를 1부터 n까지로 설정
  • 반복문 안에 조건문을 생성하여 n을 x로 나누었을 때 나머지가 1인 값을 찾음
  • min 함수를 이용해 조건에 맞는 x 값 중 가장 작은 값 출력

정답 코드

def solution(n):  

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

다른 사람 풀이

  • 조건문과 반복문을 한 줄로 작성하고 리스트 인덱싱을 이용해 가장 작은 값 출력
def solution(n):
    return [x for x in range(1,n+1) if n%x==1][0]
* 이 문제는 프로그래머스 코딩테스트 연습 1단계 문제입니다.
728x90
반응형