试题 算法提高 质数2

思路：定义一个函数判断是否为质数，然后通过for循环，是的话计数+1,且加入列表中，最后打印即可，详见代码。

资源限制
时间限制：1.0s   内存限制：256.0MB
问题描述
　　给定一个正整数N，请你输出N以内（不包含N）的质数以及质数的个数。
输入格式
　　输入一行，包含一个正整数N。
输出格式
　　共两行。
　　第1行包含一个整数，表示N以内质数的个数。
　　第2行包含若干个素数，每两个素数之间用一个空格隔开，素数从小到大输出。
样例输入
10
样例输出
4
2 3 5 7
数据规模和约定
　　N<=1000

代码：

def fun(n):
    # 判断是否为质数
    if n < 2:
        return False
    elif n == 2:
        return True
    for i in range(2,n):
        if n%i == 0:
            return False
    return True

if __name__ == "__main__":
    n = int(input())
    nums = []
    count = 0
    for i in range(n):
        if fun(i)==True:
            nums.append(i)
            count += 1
    print(count)
    for i in nums:
        print(i,end=' ')