leetcode 204 计数素数 java实现

统计所有小于非负整数 n 的质数的数量。

示例:

输入: 10
输出: 4
解释: 小于 10 的质数一共有 4 个, 它们是 2, 3, 5, 7 。

一开始我是这样写的

List list = new ArrayList();
        for(int i = 2;i < n;i++) {
        	if(isPrime(i)) {
        		list.add(i);
        	}
        }
        return list.size();
    }

	private static boolean isPrime(int i) {
		for (int j = 2; j*j <= i; j++) {
			if(i%j == 0) {
				return false;
			}
		}
		return true;
	}

但是总是超时，时间复杂度为n*根号n

后来用筛选法，去掉合数然后求素数个数
比如遍历到7，7是素数，那么14、21、28、35…就不是素数

boolean[] notPrime = new boolean[n];
		    int count = 0;
		    for (int i = 2; i < n; i++) {
		        if (notPrime[i] == false) {
		            count++;
		            for (int j = 2; i*j < n; j++) {
		                notPrime[i*j] = true;
		            }
		        }
		    }

		    return count;