Leetcode3233. 统计不是特殊数字的数字数量

Every day a Leetcode

题目来源：3233. 统计不是特殊数字的数字数量

解法1：遍历

根据题意，只有质数的平方才是特殊数字。

代码：

/*
 * @lc app=leetcode.cn id=3233 lang=cpp
 *
 * [3233] 统计不是特殊数字的数字数量
 */

// @lc code=start
class Solution
{
public:
    int nonSpecialCount(int l, int r)
    {
        int cnt = 0;
        for (int i = l; i <= r; i++)
            if (!isSpecial(i))
                cnt++;
        return cnt;
    }
    // 辅函数
    bool isSpecial(int x)
    {
        if ((int)sqrt(x) * (int)sqrt(x) == x)
        {
            return isPrime((int)sqrt(x));
        }
        return false;
    }
    bool isPrime(int x)
    {
        if (x == 1)
            return false;
        for (int i = 2; i * i <= x; i++)
            if (x % i == 0)
                return false;
        return true;
    }
};
// @lc code=end

结果：

解法2：埃式筛 + 计数

正难则反，统计区间 [l,r] 内有多少个特殊数字。

这等价于区间 [0, r] 内的特殊数字个数，减去区间 [0, l−1] 内的特殊数字个数。

代码：

/*
 * @lc app=leetcode.cn id=3233 lang=cpp
 *
 * [3233] 统计不是特殊数字的数字数量
 */

// @lc code=start
const int MX = 31622; // floor(sqrt(10^9))
bool inited = false;
vector<int> pi(MX + 5, 0);

// 埃氏筛 O(Mlog(logM))
void init()
{
    if (inited)
        return;
    for (int i = 2; i <= MX; i++)
    {
        if (pi[i] == 0)
        { // i 没有被标记，i 是质数
            pi[i] = pi[i - 1] + 1;
            for (int j = i * i; j <= MX; j += i)
            { // 标记 i 的倍数为合数
                pi[j] = -1;
            }
        }
        else
        {
            pi[i] = pi[i - 1];
        }
    }
    inited = true;
}

class Solution
{
public:
    int nonSpecialCount(int l, int r)
    {
        init();
        return r - l + 1 - (pi[(int)sqrt(r)] - pi[(int)sqrt(l - 1)]);
    }
};

// @lc code=end

结果：

复杂度分析：

时间复杂度：O(1)。不计入预处理的时间。

空间复杂度：O(1)。不计入预处理的空间。