洛谷P3172 [CQOI2015]选数(莫比乌斯反演+杜教筛)

嘛,主要是为了学杜教筛
然后发现其实可以背板子?
莫比乌斯函数前缀和
μ ( i ) = 1 − ∑ j = 2 n μ ( [ i j ] ) \mu(i)=1-\sum_{j=2}^n\mu([\frac{i}{j}]) μ(i)=1j=2nμ([ji])
欧拉函数前缀和
ϕ ( i ) = i ∗ ( i + 1 ) 2 − ∑ j = 2 n ϕ ( [ i j ] ) \phi(i)=\frac{i*(i+1)}{2}-\sum_{j=2}^n\phi([\frac{i}{j}]) ϕ(i)=2i(i+1)j=2nϕ([ji])

那么就假装会用杜教筛求比较大的莫比乌斯函数的前缀和了

这个时候愉快地用莫比乌斯反演推出公式
a n s = ∑ i = 1 h / k μ ( i ) f ( i ) ans=\sum_{i=1}^{h/k}\mu(i)f(i) ans=i=1h/kμ(i)f(i)
f ( i ) = ( h i − l − 1 i ) n f(i)={(\frac{h}{i}-\frac{l-1}{i})}^n f(i)=(ihil1)n
整除分块一波就可以过了

代码如下:

#include
#define N 2000010
#define mod 1000000007
using namespace std;

int vis[N],p[N],mu[N];
int n,k,g,h,cnt;
map<int,int> m;

int init()
{
	vis[1]=1;
	mu[1]=1;
	for(int i=2;i<N;i++)
	{
		if(!vis[i])
		{
			mu[i]=-1;
			p[++cnt]=i;
		}
		for(int j=1;j<=cnt;j++)
		{
			if(p[j]*i>=N) break;
			vis[i*p[j]]=1;
			if(!(i%p[j]))
			{
				mu[i*p[j]]=0;
				break;
			}
			else
			{
				mu[i*p[j]]=-mu[i];
			}
		}
	}
}

long long kasumi(long long a,long long b)
{
	long long ans=1ll;
	while(b)
	{
		if(b&1) ans=a*ans%mod;
		a=a*a%mod;
		b>>=1;
	}
	return ans;
}

long long solve(long long x)
{
	if(x<N) return mu[x];
	if(m.count(x)) return m[x];
	long long res=0,lim=sqrt(x);
	for(int i=2;x/i>lim;i++)
	{
		res+=solve(x/i);
	}
	for(int i=lim;i>=1;i--)
	{
		res+=(x/i-x/(i+1))*solve(i);
	}
	m[x]=1-res;
	return 1-res;
}

int main()
{
	init();
	for(int i=1;i<N;i++)
	{
		mu[i]+=mu[i-1];
	}
	scanf("%d%d%d%d",&n,&k,&g,&h);
	g--;
	long long ans=0;
	g/=k;
	h/=k;
	int l,r;
	for(l=1,r;l<=max(g,h);l=r+1)
	{
		if(g<l) break;
		r=min(h/(h/l),g/(g/l));
		ans+=1ll*(solve(r)-solve(l-1)) *kasumi(h/l-g/l,n)%mod;
		ans=(ans+mod)%mod;
	}
	for(;l<=h;l=r+1)
	{
		r=h/(h/l);
		ans+=1ll*(solve(r)-solve(l-1))*kasumi(h/l,n)%mod;
		ans=(ans+mod)%mod;
	}
	printf("%lld\n",ans);
}

你可能感兴趣的:(洛谷,数论,莫比乌斯反演,杜教筛)