【HDU6537】Neko and function（min_25筛）（二项式反演）

传送门

---

题解：

$f$有一个令人窒息的限制，就是每个位置不能是 1，我们考虑构造一个函数 $g(n.k)$ 其定义和 $f$ 一样但是允许某些位置为 $1$。

我们考虑 $k$ 个位置中有多少个 $1$，不难发现 $g$ 是 $f$ 乘上一个组合数的前缀和 ：

$$g(n,k)=\sum _{i=0}^k\left(\genfrac{}{}{0ex}{}{k}{i}\right)f(n,i)$$

于是可以用 $g$ 二项式反演得到 $f$ ：$$f(n,k)=\sum _{i=0}^k(-1{)}^{k-i}(\genfrac{}{}{0ex}{}{k}{i})g(n,i)$$

容易发现 $g$ 的计算只需要考虑各个质因子的摆放即可，换句话说， $g$ 是个积性函数。

质数点值 ：$g(p,k)=k,g(p^t,k)=\left(\genfrac{}{}{0ex}{}{t+k-1}{k-1}\right)$，于是 min_25 一次预处理算出质数个数，然后来 $k$ 次dfs计算一下 $g$ 的前缀和，二项式反演即可得到 $f$ 的前缀和。

---

代码：

#include<bits/stdc++.h>
#define ll long long
#define re register
#define cs const

using std::cerr;
using std::cout;

cs int mod=1e9+7;
inline int add(int a,int b){return a+b>=mod?a+b-mod:a+b;}
inline int dec(int a,int b){return a-b<0?a-b+mod:a-b;}
inline int mul(int a,int b){ll r=(ll)a*b;return r>=mod?r%mod:r;}
inline void Inc(int &a,int b){a+=b-mod;a+=a>>31&mod;}
inline void Dec(int &a,int b){a-=b;a+=a>>31&mod;}
inline void Mul(int &a,int b){a=mul(a,b);}
inline int MD(int x){return x>=mod?x%mod:x;}

cs int K=1e2+7;
int C[K][K];
void init_C(){
	for(int re i=0;i<K;++i){
		C[i][0]=C[i][i]=1;
		for(int re j=1;j<i;++j)
			C[i][j]=add(C[i-1][j],C[i-1][j-1]);
	}
}

int n,k;

cs int N=5e4+7;

int p[N],pc;
bool mrk[N];
void linear_sieve(){
	for(int re i=2;i<N;++i){
		if(!mrk[i])p[++pc]=i;
		for(int re j=1;i*p[j]<N;++j){
			mrk[i*p[j]]=true;
			if(i%p[j]==0)break;
		}
	}
}

int nn;
int f1[N],f2[N];

void min_25_sieve(){
	nn=sqrt(n);
	for(int re i=1;i<=nn;++i){
		f1[i]=i-1;
		f2[i]=MD(n/i-1);
	}
	for(int re i=1;p[i]<=nn;++i){
		int p=::p[i],_f=f1[p-1];
		int w1=nn/p,w2=std::min(nn,n/p/p);
		for(int re i=1;i<=w1;++i)
			Dec(f2[i],dec(f2[i*p],_f));
		for(int re i=w1+1;i<=w2;++i)
			Dec(f2[i],dec(f1[n/i/p],_f));
		for(int re i=nn;i>=(ll)p*p;--i)
			Dec(f1[i],dec(f1[i/p],_f));
	}
}

int solve(int n,int i,int k){
	int ans=mul(dec(n<=nn?f1[n]:f2[::n/n],i-1),k);
	for(int re j=i;(ll)p[j]*p[j]<=n;++j){
		for(ll now=p[j],t=1;now*p[j]<=n;now*=p[j],++t){
			Inc(ans,mul(solve(n/now,j+1,k),C[k+t-1][k-1]));
			Inc(ans,C[k+t][k-1]);
		}
	}return ans;
}

void Main(){
	init_C();linear_sieve();
	while(~scanf("%d%d",&n,&k)){
		min_25_sieve();int ans=0;
		for(int re i=0;i<k;++i)
			Inc(ans,mul(i&1?mod-C[k][i]:C[k][i],solve(n,1,k-i)));
		cout<<ans<<"\n";
	}
}

inline void file(){
#ifdef zxyoi
	freopen("function.in","r",stdin);
#endif
}
signed main(){file();Main();return 0;}