From d0fec459aba203c8f8f97acbc4224094213a75f4 Mon Sep 17 00:00:00 2001 From: Udhay Prakash Pethakamsetty Date: Sat, 2 Nov 2019 19:56:06 +0530 Subject: [PATCH] optimized --- Random_Py_scripts/check_primes.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/Random_Py_scripts/check_primes.py b/Random_Py_scripts/check_primes.py index 53af977..8bb51e5 100644 --- a/Random_Py_scripts/check_primes.py +++ b/Random_Py_scripts/check_primes.py @@ -1,3 +1,4 @@ +#!/usr/bin/python """Script to return if a number is prime""" def check_primes(num_to_check): @@ -6,11 +7,9 @@ def check_primes(num_to_check): """ count = 0 for i in range(2, num_to_check): - if num_to_check%i == 0: - count = count+1 - if count == 0: - return 1 - return 0 + if num_to_check % i == 0: + return 0 + return 1 if __name__ == "__main__": NUM = int(input())