[프로그래머스][LV.1] 나머지가 1이 되는 수 찾기 | python3

 

문제 링크: 나머지가 1이 되는 수 찾기

 

문제 설명

문제 설명

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

제한사항
- 3 ≤ n ≤ 1,000,000

 

---

문제 풀이

작은 수부터 1씩 증가시키며 나머지가 1이 되는 수를 찾는다.
while문을 돌려서 가장 먼저 나머지가 1이 되는 값이 나타나면 break로 while문을 끝낸다.

 

내코드

def solution(n):
    answer = 2
    
    while True:
        if n%answer == 1:
            break
        answer+=1
    return answer

 

 

---

다른사람풀이

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

이런 말들이 있어서 가지고 왔다.

next를 사용하면 바로 빠져나올 수 있다는데 next는 처음 본다.

 

next 찾아보니까 continue랑 비슷한 거 같은데 흠....

test_list = [1,2,3,4,5]
for i in range(len(test_list)):
    if i ==3:
        next
    print(i)

 이런식으로 3일때 print를 안하고 넘어가는 기능인데 next(for문) 이렇게 쓰면 뭐가 다른가??

 

gpt한테 물어봤다...

 

1. next()의 역할
next()는 이터레이터의 다음 값을 가져오는 함수입니다.
만약 원하는 조건을 만족하는 값을 찾으면, 그 값을 반환하고 반복 작업을 중단합니다.
따라서 for 루프나 리스트 컴프리헨션처럼 전체를 다 순회하지 않아도 됩니다.

def solution(n):
    return next(x for x in range(1, n+1) if n % x == 1)​

동작 원리:

1. x for x in range(1, n+1) if n % x == 1:
   • 제너레이터 표현식입니다. 리스트를 생성하지 않고, 조건을 만족하는 값을 필요할 때마다 하나씩 생성합니다.
2. next():
   • 이 제너레이터에서 조건을 만족하는 첫 번째 값을 반환합니다.
   • 첫 번째 값을 찾은 시점에서 즉시 연산이 종료됩니다.

---

이런게 있는 줄은 몰랐다. 잘 기억해 놔야 겠다. 

 

def solution(n):
    i = 2
    while (n-1)%i: i += 1

    return i

이건 왜 n-1 인지 이해가 안돼서 가지고 왔다.

gpt한테 물어봄. 

이제 이해가 된다. (n-1)%i가 0이 되면 답을 찾은 것이고 while문은 false(0)가 되어 나옿 수 있구나.

 

흥미로워서 가져옴