nowcoder17394 a-贝利福斯数

链接

点击跳转

题解

这题的做法就是线性筛，注意到两个a-贝利福斯数的乘积还是a-贝利福斯数，所以这个代数系统对乘法是封闭的

在素数筛中，我们计算复杂度的时候曾经说“每个合数只被最小的素数筛掉了一次”，然而在这个a-贝利福斯数代数系统上结论不在成立，比如$a=3$，$100=4\times 25=10\times 10$，$4,10,25$都是a-贝利福斯素数，所以这里统计的时候，如果在线性筛的过程中去统计可能会算多，我的做法是对这种数字打上一个标记，最后看一下多少数字有标记即可

代码

#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
#define iinf 0x3f3f3f3f
#define linf (1ll<<60)
#define eps 1e-8
#define maxn 20'000'010
#define cl(x) memset(x,0,sizeof(x))
#define rep(i,a,b) for(i=a;i<=b;i++)
#define em(x) emplace(x)
#define emb(x) emplace_back(x)
#define emf(x) emplace_front(x)
#define fi first
#define se second
#define de(x) cerr<<#x<<" = "<<x<<endl
using namespace std;
using namespace __gnu_pbds;
typedef long long ll;
typedef pair<int,int> pii;
typedef pair<ll,ll> pll;
ll read(ll x=0)
{
    ll c, f(1);
    for(c=getchar();!isdigit(c);c=getchar())if(c=='-')f=-f;
    for(;isdigit(c);c=getchar())x=x*10+c-0x30;
    return f*x;
}
ll prime[maxn], N, a, ans;
bool mark[maxn], ok[maxn];
#define f(x) (a*(x)+1)
void shai()
{
    ll i, j;
    for(i=1;f(i)<=N;i++)
    {
        if(!mark[i])prime[++*prime]=i;
        for(j=1;j<=*prime and f(i)*f(prime[j])<=N;j++)
        {
            ll t=f(i)*f(prime[j]), x=(i+prime[j]+a*i*prime[j]);
            mark[x]=1;
            if(!mark[i])ok[x]=1;
            if(f(i)%f(prime[j])==0)break;
        }
    }
}
int main()
{
    ll i;
    a=read(), N=read();
    shai();
    rep(i,1,maxn-1)ans+=ok[i];
    printf("%lld",ans);
    return 0;
}