【HDU 5608】function(杜教筛)

最新推荐文章于 2020-09-01 21:04:48 发布

原创最新推荐文章于 2020-09-01 21:04:48 发布 · 312 阅读

1 ·

CC 4.0 BY-SA版权

数论同时被 2 个专栏收录

41 篇文章

订阅专栏

杜教筛

8 篇文章

订阅专栏

本文介绍了一种基于数论的优化求和算法，通过预处理和递归集约化方式加速计算过程。针对特定数学方程，实现了高效计算并解决大规模数据集的问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Problem Description
There is a function f(x),which is defined on the natural numbers set N,satisfies the following eqaution

$N^2−3N+2=\sum_{d|N}f(d)$

calulate $\sum_{i=1}^nf(i)\ mod\ 10^9+7$ .

Input
the first line contains a positive integer T,means the number of the test cases.

next T lines there is a number N

$T≤500,N≤10^9$

only 5 test cases has N>106.

Output
Tlines,each line contains a number,means the answer to the i-th test case.

Sample Input
1
3

Sample Output
2

$1^2-3*1+2=f(1)=0$
$2^2-3*2+2=f(2)+f(1)=0->f(2)=0$
$3^2-3*3+2=f(3)+f(1)=2->f(3)=2$
$f(1)+f(2)+f(3)=2$

先前的写法是纯记忆化，但是超时了啊。所以果然还是像杜教筛一样先预处理前1000000的ans，然后在递归集约化搞就是了。
代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<map>
#include<algorithm>
#include<cmath>
#define maxx 1000005
#define ll long long
using namespace std;
const int mod=1000000007;
ll ans[maxx];
map<ll ,ll> M;
ll inv=333333336;
void init()
{
    for(int i=1;i<maxx;i++) ans[i]=((ll)i-1)*(i-2)%mod;
    for(int i=1;i<maxx;i++)
        for(int j=i+i;j<maxx;j+=i)
            ans[j]=(ans[j]-ans[i]+mod)%mod;
    for(int i=1;i<maxx;i++)
        ans[i]=(ans[i]+ans[i-1])%mod;
}
ll work(ll x)
{
    if(x<maxx)  return ans[x];
    if(M[x])return M[x];
    ll res=x*(x-1)%mod*(x-2)%mod*inv%mod;
    for(ll i=2,last;i<=x;i=last+1)
    {
        last=x/(x/i);
        res=(res-(last-i+1)*work(x/i)%mod)%mod;
    }
    res=(res+mod)%mod;
    M[x]=res;
    return res;
}
ll p(ll a,ll b)
{
    ll ans=1;
    while(b)
    {
        if(b&1)ans=ans*a%mod;
        a=a*a%mod;
        b>>=1;
    }
    return ans;
}
int main()
{
    //cout<<p(3,mod-2)<<endl;
    init();
    ans[1]=0;
    int t;
    cin>>t;
    while(t--)
    {
        ll n;
        scanf("%lld",&n);
        printf("%lld\n",work(n));
    }
    return 0;
}