P2398 GCD SUM(莫比乌斯反演)

最新推荐文章于 2023-01-26 16:10:19 发布

_九磅十五便士_

最新推荐文章于 2023-01-26 16:10:19 发布

阅读量116

点赞数 1

分类专栏： # 数论文章标签：迪利克雷卷积素数筛法欧拉函数算法效率复杂度优化

本文链接：https://blog.csdn.net/m0_51841071/article/details/116207077

版权

数论专栏收录该内容

42 篇文章 1 订阅

订阅专栏

题目传送门

前置知识：
本题应用到的迪利克雷卷积
在这里插入图片描述
推导过程如下：

因此用O(n)的算法就可以AC本题

#include<iostream>
using namespace std;
typedef long long ll;
const int N=1e5+10;
int primes[N], cnt;     // primes[]存储所有素数
int euler[N];           // 存储每个数的欧拉函数
bool st[N];         // st[x]存储x是否被筛掉
void eulers(int n)
{
    euler[1] = 1;
    for (int i = 2; i <= n; i ++ )
    {
        if (!st[i])
        {
            primes[cnt ++ ] = i;
            euler[i] = i - 1;
        }
        for (int j = 0; primes[j] <= n / i; j ++ )
        {
            int t = primes[j] * i;
            st[t] = true;
            if (i % primes[j] == 0)
            {
                euler[t] = euler[i] * primes[j];
                break;
            }
            euler[t] = euler[i] * (primes[j] - 1);
        }
    }
}
int main(){
	eulers(N-10);
	int n;
	cin>>n;
	ll ans=0;
	for(int i=1;i<=n;i++){
		ans+=(ll)euler[i]*(ll)(n/i)*(n/i);
	}
	cout<<ans<<endl;
}

当然，还可以用数论分块，将复杂度优化到O(n^1/2)

#include<iostream>
using namespace std;
typedef long long ll;
const ll N=1e5+10;
ll primes[N], cnt;     // primes[]存储所有素数
ll euler[N];           // 存储每个数的欧拉函数
bool st[N];         // st[x]存储x是否被筛掉
void eulers(ll n)
{
    euler[1] = 1;
    for (ll i = 2; i <= n; i ++ )
    {
        if (!st[i])
        {
            primes[cnt ++ ] = i;
            euler[i] = i - 1;
        }
        for (ll j = 0; primes[j] <= n / i; j ++ )
        {
            ll t = primes[j] * i;
            st[t] = true;
            if (i % primes[j] == 0)
            {
                euler[t] = euler[i] * primes[j];
                break;
            }
            euler[t] = euler[i] * (primes[j] - 1);
        }
    }
    for(ll i=1;i<=N;i++){
    	euler[i]+=euler[i-1];
	}
}
int main(){
	eulers(N-10);
	ll n;
	cin>>n;
	ll ans=0;
	ll l=1,r;
	for(l=1;l<=n;l=r+1){
		r=n/(n/l);
		ans+=(euler[r]-euler[l-1])*(n/l)*(n/l);
	}
	cout<<ans<<endl;
}