【Luogu5348】密码解锁（莫比乌斯反演，数论）
题面
洛谷
题解
首先题目给定的限制是\(\sum_{n|i}a[i]=\mu(n)\)，然后把这个东西反演一下，
莫比乌斯反演的式子是：\(g(n)=\sum_{n|i}f(i)\rightarrow f(n)=\sum_{n|i}g(i)\mu(\frac{i}{n})\)，在这里\(\mu\)就是\(g\)，而\(a\)就是\(f\)。
所以我们可以得到：\(a[m]=\sum_{m|i}\mu(i)\mu(\frac{i}{m})=\sum_{i=1}^{n/m}\mu(i)\mu(im)\)。
然后直接把后面拆开，得到：\(\mu(m)\sum_{i=1}^{n/m}[gcd(i,m)=1]\mu(i)^2\)
后面那一半接着拆，可以得到：
\[\begin{aligned} a[m]&=\mu(m)\sum_{i=1}^{n/m}\mu(i)^2\sum_{j|i,j|m}\mu(j)\\ &=\mu(m)\sum_{j|m}\mu(j)\sum_{j|i}^{n/m}\mu(i)^2 \end{aligned}\]
前面的\(j\)显然只有\(\sqrt m\) 个了。
后面一半枚举最小的平方因子，然后把这部分的贡献减去就行了，这部分的复杂度是\(O(\sqrt \frac{n}{m})\)。
所以总的复杂度就是\(O(\sigma_0(m)\sqrt{\frac{n}{m}})\)。
#include
#include
#include
using namespace std;
#define ll long long
inline ll read()
{
ll x=0;bool t=false;char ch=getchar();
while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
if(ch=='-')t=true,ch=getchar();
while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
return t?-x:x;
}
ll n,ans;int m,fac[100],p;;
const int N=1e6;
bool zs[N];
int mu[N],pri[N],tot;
void Sieve()
{
mu[1]=1;
for(int i=2;i1){fl=true;break;}
}
if(fl){puts("0");continue;}
if(x>1)fac[++p]=x;
dfs(1,1,1);
printf("%lld\n",ans*((p&1)?-1:1));
}
}