除数函数求和 1 （线性筛素数应用+数论分块）

oj 1599

Description

 

Input

第一行两个正整数 n , k 。

对于 100% 的数据，1 ≤ n , k ≤ 107 。

Output

第一行输出答案。

Sample Input

5 2

Sample Output

63

Author

John

 

参考题解：http://www.cnblogs.com/five20/p/9226187.html

直接枚举显然不行，我们考虑改为求n的某一因子d在整个函数中的贡献是多少。

　　套上数论分块的思想，一个因子d对式子的贡献是。

　　这样我们需要处理的就是，直接快速幂求出每个因子的幂是肯定不行的，因为n是1e7，直接会T。

　　那么还是考虑优化，我们发现，每个数都能唯一分解，而在求幂时会有重复计算的质因子幂。于是，我们考虑线筛，这样就可以用每个数的最小质因子幂去算它的幂了，那么整个过程只会对n≤1e7内的质数进行快速幂，最后复杂度就成了，完全可行。

　　所以最后就只需再O(n)扫一遍因子累加贡献求和就好了。

 

///参考网上题解http://www.cnblogs.com/five20/p/9226187.html

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;

typedef long long LL;
const LL mod=1e9+7;
const int maxn=1e7+10;

LL sum[maxn];
bool isprime[maxn];
int prime[maxn];
int n,k;

LL fastpow(LL i,LL k) ///快速幂
{
    LL ans=1;

    while(k)
    {
        if(k&1) ans=ans*i%mod;
        k>>=1;
        i=i*i%mod;
    }

    return ans;
}
void init()
{
    sum[1]=1;
    int cnt=0;
    for(int i=2;i<=n+2;i++) ///线性筛素数
    {
        if(!isprime[i]) {
            prime[++cnt]=i;
            sum[i]=fastpow((LL)i,(LL)k);///存储每个因子的幂
        }

        for(int j=1;j<=cnt&&prime[j]*i<=n+2;j++)
        {
            isprime[i*prime[j]]=true;

            sum[prime[j]*i]=sum[i]*sum[prime[j]]%mod;
            ///顺便用每个数的最小质因子幂去算它的幂
            ///例如sum[14]可以由sum[7]*sum[2]来求 就相当于7^k*2^k=(7*2)^K,
            
            ///因为我们只是要求sum[14]的值，管它由谁组成，例如sum[14]=sum[3]*sum[4]求得，
            ///而刚好线性筛素数是每个合数只会被最小质因子筛掉
            ///所以我们就可以利用这个特性来求sum的值
            
            ///这样就很完美的线性筛素数的同时也把因子的幂筛好了
            if(i%prime[j]==0) break;
        }
    }
}
int main()
{
    while(~scanf("%d%d",&n,&k))
    {
        memset(isprime,false,sizeof(isprime));
        init();

        LL ans=0;


        for(int i=1;i<=n;i++) ///枚举每个因子i的幂，例如因子2的数有n/2个，因子3的数有n/3个
        {
            ans=(ans+(LL)(n/i)*sum[i])%mod;///加上每个因子的贡献值

//            printf("sum[%d]=%d\n",i,sum[i]);
        }

        printf("%lld\n",ans);
    }

    return 0;
}

/// 1 4 9 16 25
/// 10000000 10000000
/// 674327231