def solution(n):
for i in range(101):
if ((i * 6) // n > 0) & ((i * 6) % n == 0):
return i
입출력 예시
에 맞춰서 수학 문제 풀듯이 해결했다.
range 함수 부분은 코드를 작성하면서도 이게 최선인가 싶었다.
def solution(n):
i=1
while(1):
if (6*i)%n==0:
return i
i+=1
내가 거슬렸던 부분을 시원하게 해결해준 풀이다.
range 함수로 범위를 설정하는 대신, while문으로 무한루프를 구현했다. 나머지가 0이 될 때까지 i를 1씩 더해가며 조건에 부합하는 피자 수를 리턴한다.
또한, i=1로 초기화함으로써 몫이 0인 경우를 방지하였다.