oj 1599
Description
Input
第一行两个正整数 n , k 。
对于 100% 的数据,1 ≤ n , k ≤ 107 。
Output
第一行输出答案。
Sample Input
5 2
Sample Output
63
Author
参考题解: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