luoguP5325 【模板】Min_25筛

题目描述

数据范围

题解

话说打出min25板子之后真的这些题都是切得极爽。

对着板子改两下就好了。233。
这题版题，直接上min25即可。
水博客量

代码

#include 
#include 
#include 
#include 
using namespace std;
const long long mo=1000000007;
const int maxn=1000010;

long long zs[maxn],n,m,d[maxn],sum[maxn],sum1[maxn],id,wz1,wz2,id1[maxn],id2[maxn],g2[maxn],g1[maxn],nex,now;
int tot;
bool bz[maxn];

long long qsm(long long a,long long b)
{
	long long t=1;
	long long y=a;
	while (b>0)
	{
		if ((b&1)==1) t=t*y%mo;
		y=y*y%mo;
		b/=2;
	}
	return t;
}

void getg()
{
	for (int i=1;i<=zs[0];i++)
	{
		for (int j=1;j<=tot;j++)
		{
			if (zs[i]*zs[i]<=d[j])
			{
				id=d[j]/zs[i];
				if (id<=m) wz1=id1[id];else wz1=id2[n/id];
				id=zs[i]-1;
				if (id<=m) wz2=id1[id];else wz2=id2[n/id];
				
				g2[j]=(g2[j]-zs[i]*zs[i]%mo*(g2[wz1]-sum1[i-1]+mo)%mo+mo)%mo;
				g1[j]=(g1[j]-zs[i]%mo*((g1[wz1]-sum[i-1]+mo)%mo)%mo+mo)%mo;
			}
			else
			{
				break;
			}
		}
	}
}

long long gets(long long a,int b)
{
	if (a==0 || zs[b]>a) return 0;
//	printf("%lld\n",a);
	long long id=a,wz1,wz2;
	if (id<=m) wz1=id1[id]; else wz1=id2[n/id];
	
	long long ss=0;
	ss=((g2[wz1]-g1[wz1])-(sum1[b-1]-sum[b-1])%mo+mo)%mo;
	
	for (int i=b;i<=zs[0];i++)
	{
		if (zs[i]*zs[i]<=a)
		{
			long long mi=zs[i];
			if (zs[i]*zs[i]>a) continue;
			for (int e=1;mi*zs[i]<=a;e++,mi=mi*zs[i])
			{
				ss=(ss+(qsm(zs[i],e)*(qsm(zs[i],e)-1)%mo)*gets(a/mi,i+1)%mo+(qsm(zs[i],e+1)*(qsm(zs[i],e+1)-1)%mo)%mo)%mo;
			}
		}
		else break;
	}
	return ss;
}

int main()
{
//	freopen("data.in","r",stdin);
//	freopen("data.out","w",stdout);
	for (int i=2;i<=100000;i++)
	{
		if (!bz[i])
		{
			zs[0]++;
			zs[zs[0]]=i;
			for (int j=1;j<=100000/i;j++)
			{
				bz[j*i]=true;
			}
		}
	}
	
	for (int i=1;i<=zs[0];i++)
	{
		sum[i]=(sum[i-1]+zs[i])%mo;
		sum1[i]=(sum1[i-1]+zs[i]*zs[i]%mo)%mo;
	}
	
	scanf("%lld",&n);
	if (n==1)
	{
		printf("1\n");
		return 0;
	}
	
	m=sqrt(n);
	for (long long i=1;i<=n;i=nex+1)
	{
		if (i>100000)
		{
			int j=0;
		}
		nex=n/(n/i);
		d[++tot]=n/i;
		now=(n/i)%mo;
		g2[tot]=(now*(now+1)%mo*(2*now+1)%mo*qsm(6,mo-2)%mo+mo-1)%mo;
		g1[tot]=(((now+1)%mo)*(now%mo)%mo*qsm(2,mo-2)%mo-1+mo)%mo;
		if (d[tot]<=m) id1[n/i]=tot;
		else id2[n/(n/i)]=tot;
	}
	
	getg();
	
	long long ans=gets(n,1);
	printf("%lld\n",(ans+1)%mo);
}