ACM训练联盟周赛第一场 C-A Simple Math Question(筛+贡献+卡常)

题目

题解

非常卡常的题，只有16384KB

bitset占用空间是bool的八分之一，优化一波

然后3e8缩到1e8，特判2和3的倍数，把剩下的压缩

剩下的就是裸的勒让德定理枚举贡献的统计了

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<bitset> 
using namespace std;
const int maxm=1e6+10;
const int maxn=1e8+10;
bitset<maxn>ok;
unsigned int prime[maxm],cnt;
unsigned int n,a[4],res;
unsigned int tmp,num,now,v;
void add(int i,int n)
{
	tmp=n,num=0,now=1,v=0;
	for(;tmp;tmp/=i)
	num+=tmp/i;
	for(int j=3;j>=0;--j)
	v+=a[j]*now,now*=i;
	res+=num*v;
}
void sieve(int n)
{ 
    res=0;
    prime[cnt++]=2;add(2,n); 
    prime[cnt++]=3;add(3,n);
	for(int i=5;i<=n;)
	{
		if(i%2&&i%3)
        {
        	 if(!ok[i/6*2+(i%6==5)])
             {
             add(i,n);
             if(cnt<maxm)prime[cnt++]=i;
             }
        }
		for(int j=0;j<cnt;++j)
		{
			if(i>n/prime[j])break;
			int tmp=i*prime[j];
            if(tmp%2==0||tmp%3==0)continue;
            ok[tmp/6*2+(tmp%6==5)]=1;
			if(i%prime[j]==0)break;
		}
        i%6==5?i+=2:i+=4;
	}
}
int main()
{
	scanf("%u",&n);
	for(int i=0;i<=3;++i)
	scanf("%u",&a[i]);
	sieve(n);
	printf("%u\n",res);
	return 0;
}