[COS PRO 2급-Python] 소수의 갯수 구하기

[COS PRO 2급-Python] 짝수들의 제곱의 합 구하기

문제 설명

자연수의 각 자릿수 중에서 소수는 몇 개인지 구하려 합니다.
즉, 자연수를 각 자릿수별로 나누었을 때, 2, 3, 5, 7이 몇 개 있는지 구하려 합니다.


예를 들어, 자연수가 29022531일 때, 각 자릿수 중 소수의 위치는 다음과 같습니다.

• 2 9 0 2 2 5 3 1

따라서 소수의 개수는 총 5개입니다.
자연수 number가 매개변수로 주어질 때, number의 각 자릿수 중 소수는 몇 개인지 return 하도록 solution 함수를 작성했습니다.
그러나, 코드 일부분이 잘못되어있기 때문에, 몇몇 입력에 대해서는 올바르게 동작하지 않습니다.
주어진 코드에서 한 줄만 변경해서 모든 입력에 대해 올바르게 동작하도록 수정하세요.

매개변수 설명

자연수 number가 solution 함수의 매개변수로 주어집니다.

• number는 1이상 1,000,000,000 이하의 자연수 입니다.

• N과 M은 1 이상 1,000 이하의 자연수이며, N < M을 항상 만족합니다.

return 값 설명

number의 각 자릿수 중 소수는 몇 개인지 return 해주세요.

예시

알고리즘 구상

▶ while문에 number가 0보다 클때만 돌려야 한다!

내가 쓴 코드!

def solution(number):
	count = 0
	while number > 0:
		n = number % 10
		if n == 2 or n == 3 or n == 5 or n == 7:
			count += 1
		number //= 10
	return count

느낀점

화려한 문법과 코드에 현혹되지 말고 처음부터 차근차근 읽어보자!