题目
GJJ每天都要膜拜一发数学大佬,因为GJJ的数学太差了。这不,GJJ又遇到难题了,他想求助WJJ,但是WJJ这几天忙于追妹子,哪有时间给他讲题, 于是GJJ求助于热爱ACM的你,Acmer们能帮帮他吗?问题是求: k^0 + k^1 +…+ k^(n) mod p (0 < k < 100, 0 <= n <= 10^9, p = 1000000007)
例如:6^0 + 6^1 +…+ 6^(10) mod 1000000007 (其中k = 6, n = 10, p = 1000000007)
Input
输入测试数据有多组,每组输入两个整数k, n
Output
每组测试数据输出:Case #: 计算结果
Sample Input
2 1
6 10
Sample Output
Case 1: 3
Case 2: 72559411
HINT
Source
这道题当初没写出来,也没想到是等比数列,还是思维发散不行,还有一点就是在取模的时候有个除数为了精确就可以求乘法逆元转化为乘法来做
若对于数字A,C 存在X,使A * X = 1 (mod C) ,那么称X为 A 对C的乘法逆元。
乘法逆元怎么说呢,例如 :
12 / 4 mod 7 = ? 很显然结果是3
我们现在对于数对 (4,7), 可以知道 X = 2是 4 对7的乘法逆元即2*4=1(mod 7)
那么我们有(12 / 4) * (4 * 2 ) = (?) * (1) (mod 7)
除法被完美地转化为了乘法从而避免精度丢失
这里只说当 a p互质时(就是 a p最大公约数为1)
求法 一 费马小定理
如果a p互质 a^(p-1) mod p = 1
a的乘法逆元就是 a^(p-2)
算法时间复杂度O(log2 n) 就是个快速幂
求法二 扩展欧几里得
ax= 1mod p;
可以写成
ax - py =1
这就看出来了吧 时间复杂度 O(ln n)
那这道题就好写了
一 费马小定理
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<set>
#include<map>
#include<stack>
#include<queue>
#include<string>
#include<cstring>
#include<cmath>
#include<sstream>
#include<cstdlib>
using namespace std;
#define inf 0x3f3f3f3f
long long mod = 1000000007;
long long poww(long long a,long long b)
{
long long ans=1;
while(b)
{
if(b&1) ans=ans*a%mod;
b>>=1;
a=a*a%mod;
}
return ans;
}
int main()
{
int k,n;
int tt=1;
while(cin>>k>>n)
{ cout<<"Case "<<tt++<<": ";
if(k==1)
{
cout<<(n+1)%mod<<endl;continue;
}
long long kk=poww(k,n+1)-1;
long long nn=k-1;
int ans=poww(nn,mod-2); 费马小定理
cout<<ans*kk%mod<<endl;
}
return 0;
}
二 扩展欧几里得
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<set>
#include<map>
#include<stack>
#include<queue>
#include<string>
#include<cstring>
#include<cmath>
#include<sstream>
#include<cstdlib>
using namespace std;
#define inf 0x3f3f3f3f
long long mod = 1000000007;
void exgcd(int a,int b,long long &gg,long long &x,long long &y)
{
if(!b) gg=a,x=1,y=0;
else
{
exgcd(b,a%b,gg,y,x);
y-=x*(a/b);
}
}
int main()
{
int k,n;
int tt=1;
while(cin>>k>>n)
{ cout<<"Case "<<tt++<<": ";
if(k==1)
{
cout<<(n+1)%mod<<endl;continue;
}
long long kk=poww(k,n+1)-1;
long long nn=k-1;
long long gg,x,y; //扩展欧几里得 gg 公约数 ax+by=z;
exgcd(nn,mod,gg,x,y);
long long ans=(x%mod+mod)%mod;
cout<<ans*kk%mod<<endl;
}
return 0;
}