P5495 【模板】Dirichlet 前缀和

最新推荐文章于 2022-09-07 08:56:49 发布

_Shmily

最新推荐文章于 2022-09-07 08:56:49 发布

阅读量94

点赞数

分类专栏：数论基本算法

本文链接：https://blog.csdn.net/zhaoxinxin1234/article/details/104227362

版权

数论基本算法专栏收录该内容

67 篇文章 0 订阅

订阅专栏

bk = sum of ai if i | k

维护一个质因子分解之后，关于指数的前缀和。
理论时间复杂度O（nloglogn）

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
#include<set>
#include<vector>
#define ll long long
#define llu unsigned ll
using namespace std;

#define uint unsigned int
uint seed;
inline uint getnext(){
	seed^=seed<<13;
	seed^=seed>>17;
	seed^=seed<<5;
	return seed;
}

const int maxn=20000100;
int prime[maxn],cnt=0,n;
bool ha[maxn];
uint a[maxn];

void Prime(void)
{
    ha[1]=true;
    for(int i=2;i<=n;i++)
    {
        if(!ha[i]) prime[++cnt]=i;
        for(int j=1;j<=cnt&&prime[j]*i<=n;j++)
        {
            ha[prime[j]*i]=true;
            if(i%prime[j]==0) break;
        }
    }
}

int main(void)
{
    scanf("%d%u",&n,&seed);
    Prime();
    for(int i=1;i<=n;i++)
        a[i]=getnext();
    for(int i=1;i<=cnt;i++)
        for(int j=1;j*prime[i]<=n;j++)
            a[j*prime[i]]+=a[j];
    uint ans=0;
    for(int i=1;i<=n;i++)
        ans^=a[i];
    printf("%u\n",ans);
    return 0;
}