298 · 寻找素数

描述
输出n以内所有的素数。

保证 n 是100以内的整数。

样例
样例 1:

输入：5
输出：[2, 3, 5]

class Solution:
    """
    @param n: an integer
    @return: return all prime numbers within n.
    """
    def prime(self, n):
        flags = [True for i in range(n+1)]
        for i in range(2,int(n**0.5)+1):
            for j in range(i*2,n+1,i):
                flags[j] = False
        return [i for i in range(2,n+1) if flags[i]==True]