hdu 5608 function

最新推荐文章于 2020-08-19 22:42:58 发布

sdfzyhx

最新推荐文章于 2020-08-19 22:42:58 发布

阅读量382

点赞数

分类专栏：数学 hdu

本文链接：https://blog.csdn.net/sdfzyhx/article/details/72845975

版权

数学同时被 2 个专栏收录

229 篇文章 0 订阅

订阅专栏

hdu

36 篇文章 0 订阅

订阅专栏

= = = s (n) \sum i = 1 n f (n) \sum i = 1 n (i 2 - 3 i + 2) - \sum i = 1 n \sum d | i, d < i f (d) n ( n + 1 ) ( 2 n + 1 ) 6 - 3 n ( n + 1 ) 2 + 2 n - \sum i = 2 n \sum d = 1 ⌊ n i ⌋ f (d)

$\begin{align} & s(n) \\ = & \sum_{i=1}^n f(n) \\ = & \sum_{i=1}^n(i^2-3i+2)-\sum_{i=1}^n\sum_{d|i,d<i} f(d) \\ = & \frac{n(n+1)(2n+1)}6-\frac{3n(n+1)}2+2n-\sum_{i=2}^n\sum_{d=1}^{\lfloor\frac n i \rfloor} f(d) \end{align}$
杜教筛即可。预处理的时候做不到线性了，但是仍然可以

O(klogk) $O(k\log k)$ 做到。

#include<cstdio>
#include<algorithm>
using namespace std;
#define LL long long
const int maxn=800000,maxm=3000000,p=1000000007,mod=1000007,inv2=500000004,inv6=166666668;
int f[maxn+10],sum[maxn+10],fir[mod+10],ne[maxm],val[maxm],ans[maxm],tot;
int inc(int x,int y)
{
    x+=y;
    return x>=p?x-p:x;
}
int dec(int x,int y)
{
    x-=y;
    return x<0?x+p:x;
}
int get(int n)
{
    return inc(dec((LL)n*(n+1)%p*(2*n+1)%p*inv6%p,3LL*n%p*(n+1)%p*inv2%p),2*n%p);
}
int find(int n)
{
    for (int i=fir[n%mod];i;i=ne[i])
        if (val[i]==n) return ans[i];
    return -1;
}
void ins(int n,int x)
{
    int y=n%mod;
    tot++;
    ne[tot]=fir[y];
    fir[y]=tot;
    val[tot]=n;
    ans[tot]=x;
}
int calc(int n)
{
    if (n<=maxn) return sum[n];
    int ret=find(n);
    if (ret>=0) return ret;
    ret=get(n);
    for (int i=2,j;i<=n;i=j+1)
    {
        j=n/(n/i);
        ret=dec(ret,(LL)(j-i+1)*calc(n/i)%p);
    }
    ins(n,ret);
    return ret;
}
int main()
{
    for (int i=1;i<=maxn;i++)
    {
        f[i]=inc(f[i],inc(dec((LL)i*i%p,3*i%p),2));
        sum[i]=inc(sum[i-1],f[i]);
        for (int j=i+i;j<=maxn;j+=i)
            f[j]=dec(f[j],f[i]);
    }
    int T,n;
    scanf("%d",&T);
    while (T--)
    {
        scanf("%d",&n);
        printf("%d\n",calc(n));
    }
}