力扣 204. 計數質數(每日一題)

雲軒x發表於2020-12-03

統計所有小於非負整數 n 的質數的數量。

示例 1:

輸入:n = 10
輸出:4
解釋:小於 10 的質數一共有 4 個, 它們是 2, 3, 5, 7 。
示例 2:

輸入:n = 0
輸出:0
示例 3:

輸入:n = 1
輸出:0

提示:

0 <= n <= 5 * 106

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

相關文章