7-2 素数判断（循环python）

7-2 素数判断

输入一个正整数m，判断该数是否为素数。
输入格式:

首先输入测试组数T，然后输入T组测试数据。每组测试输入一个正整数m。
输出格式:

对于每组测试，若m为素数则输出“yes”；反之输出“no”。注意：引号不必输出。
输入样例:

3
9
3
7

输出样例:

no
yes
yes

 日常写法

import math
T=int(input())
for i in range(0,T):
    m = int (input())
    k = int (math.sqrt(m))
    if (m==1):
        print("no")
    else:
        for i in range(2,k+2):
            if m % i == 0:
                break
        if i == k+1:
            print ("yes")
        else:
            print ("no")

函数写法

import math
T=int(input())
def shushu(m,k):
    if (m==1):
        print("no")
    else:
        for i in range(2,k+2):
            if m % i == 0:
                break
        if i == k+1:
            print ("yes")
        else:
            print ("no")
for i in range(0,T):
    m = int (input())
    k = int (math.sqrt(m))
    shushu(m,k)

    

注意：在判断素数中，我们在以前的思路是，如果一个数除了1和它本身不能被其它数所整除，就是素数，然后再i到n中不断尝试，现在我们可以将执行过程执行到sqrt(n)这样大大加快了代码的运行速度，以便于计算机的运行，使我们的代码运行效率更高