【HDU 5608】function(杜教筛)

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

coldfresh

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

阅读量278

点赞数

分类专栏：数论杜教筛

本文链接：https://blog.csdn.net/Coldfresh/article/details/81254199

版权

数论同时被 2 个专栏收录

41 篇文章 1 订阅

订阅专栏

杜教筛

8 篇文章 0 订阅

订阅专栏

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;
}