[소수 판정] Python으로 소수 판정하기

개념

소수: 1보다 큰 자연수 중, 1과 자기 자신만을 약수로 가지는 수

 

코드

일반 소수 판별 코드

def is_prime_number(x):
	for i in range(2, x):
		if x % i == 0:
			return False
	return True


print(is_prime_number(4)) # False
print(is_prime_number(7)) # True

 

개선된 소수 판별 코드

def is_prime_number(x):
	for i in range(2, int(x ** 0.5) + 1):
    	if x % i == 0:
			return False
	return True


print(is_prime_number(4)) # False
print(is_prime_number(7)) # True

 

Reference 

이것이 코딩테스트다 with Python