LeetCode 204 Count Primes

Problem:

Count the number of prime numbers less than a non-negative number, n.

Summary:

判断小于某非负数n的质数个数。

Solution:

用所谓的“刷质数表”的方式先用HashTable记录小于n的所有数是质数还是合数，再逐一数出。

看了题目中的Hint才知道这种方法有一个更加高大上的名字Sieve of Eratosthenes

即在每判断一个数为质数时，将它的bei'shu倍数均计为合数。

class Solution {
public:
    int countPrimes(int n) {
        if (n == 0 || n == 1) {
            return 0;
        }
        
        vector<int> prime(n, 0);
        
        prime[0] = prime[1] = 1;
        for (long long i = 2; i < n; i++) {
            if (!prime[i]) {
                for (long long j = i * i; j < n; j += i) {
                    prime[j] = 1;
                }
            }
        }
        
        int cnt = 0;
        for (int i = 0; i < n; i++) {
            if (!prime[i]) {
                cnt++;
            }
        }
        
        return cnt;
    }
};