基准时间限制:1 秒 空间限制:131072 KB 分值: 20
难度:3级算法题
求:3^0 + 3^1 +...+ 3^(N) mod 1000000007
Input
输入一个数N(0 <= N <= 10^9)
Output
输出:计算结果
Input示例
3
Output示例
40
解法一:直接利用等比数列求和公式,得到 [ (pow(3,n+1) - 1 ) / 2 ] % m。又因为m=1e9+7,是质数。然后就可以利用费马小定理+快速幂就可以求解 逆元。推导过程见 51nod 1119
代码如下:
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long LL;
const int mod = 1e9+7;
LL pow_quick(LL n,LL m)
{
LL res=1;
while(m)
{
if(m&1)
res = (res*n)%mod;
n = (n*n)%mod;
m>>=1;
}
return res;
}
void slove(LL n)
{
//n++;
LL ans=pow_quick(3,n+1)-1;
ans = (ans*pow_quick(2,mod-2)) % mod;
printf("%lld\n",ans);
}
int main()
{
LL n;
while(scanf("%lld",&n)!=EOF)
slove(n);
return 0;
}
解法二:直接利用矩阵快速幂,可以避免逆元运算。
矩阵如下: sn sn-1 3 1
= *
1 1 0 1
代码如下:
#include <cstdio>
#include <cstring>
#include <vector>
#include <algorithm>
using namespace std;
typedef long long LL;
typedef vector<LL> vec;
typedef vector<vec> mat;
const int mod = 1e9+7;
mat mul(mat &A,mat &B)
{
mat C(A.size(), vec(B[0].size()));
for(int i=0;i<A.size();++i)
{
for(int k=0;k<B.size();++k)
{
for(int j=0;j<B[0].size();++j)
C[i][j] = (C[i][j]+A[i][k]*B[k][j]) % mod;
}
}
return C;
}
mat pow_quick(mat A,LL n)
{
mat B(A.size(), vec(A.size()));
for(int i=0;i<A.size();++i)
B[i][i]=1;
while(n>0)
{
if(n&1)
B=mul(B,A);
A=mul(A,A);
n>>=1;
}
return B;
}
int main()
{
LL n;
while(scanf("%lld",&n)!=EOF){
mat A(2, vec(2));
A[0][0]=3; A[0][1]=1;
A[1][0]=0; A[1][1]=1;
A = pow_quick(A,n+1);
printf("%lld\n",A[0][1]);
}
return 0;
}