容斥原理应用(求1~r中有多少个数与n互素)

问题:求1~r中有多少个数与n互素。


对于这个问题由容斥原理,我们有3种写法,其实效率差不多。分别是:dfs,队列数组,位运算。


先说说位运算吧:

用二进制1,0来表示第几个素因子是否被用到,如m=3,三个因子是2,3,5,则i=3时二进制是011,表示第2、3个因子被用到


 

LL Solve(LL n,LL r)
{
    vector<LL> p;
    for(LL i=2; i*i<=n; i++)
    {
        if(n%i==0)
        {
            p.push_back(i);
            while(n%i==0) n/=i;
        }
    }
    if(n>1)
        p.push_back(n);
    LL ans=0;
    for(LL msk=1; msk<(1<<p.size()); msk++)
    {
        LL multi=1,bits=0;
        for(LL i=0; i<p.size(); i++)
        {
            if(msk&(1<<i))  //判断第几个因子目前被用到
            {
                ++bits;
                multi*=p[i];
            }
        }
        LL cur=r/multi;
        if(bits&1) ans+=cur;
        else       ans-=cur;
    }
    return r-ans;
}

 

然后就是dfs的实现:

 

void Solve(LL n)
{
    p.clear();
    for(LL i=2; i*i<=n; i++)
    {
        if(n%i==0)
        {
            p.push_back(i);
            while(n%i==0) n/=i;
        }
    }
    if(n>1)
        p.push_back(n);
}

void dfs(LL k,LL t,LL s,LL n)
{
    if(k==p.size())
    {
        if(t&1) ans-=n/s;
        else    ans+=n/s;
        return;
    }
    dfs(k+1,t,s,n);
    dfs(k+1,t+1,s*p[k],n);
}

//主函数内是:
dfs(0,0,1,r);


经典题目:HDU4135,HDU2841,HDU1695

 



 

你可能感兴趣的:(原理)