题目传送门:https://ac.nowcoder.com/acm/contest/10507/A
来源:牛客网
时间限制:C/C++ 1秒,其他语言2秒
空间限制:C/C++ 524288K,其他语言1048576K
64bit IO Format: %lld
题目描述
广义肥波那契数列,以递归的方法定义如下:
{f1=1f2=1fn=a⋅fn−1+b⋅fn−2(n≥3,n∈Z)\begin{cases}f_1=1\f_2=1\f_n=a\cdot f_{n-1}+b\cdot f_{n-2}(n\ge 3, n\in \mathbb Z)\end{cases}⎩⎪⎨⎪⎧f1=1f2=1fn=a⋅fn−1+b⋅fn−2(n≥3,n∈Z)
例如,当a=b=1a=b=\text 1a=b=1时,数列为[1,1,2,3,5,8,13,…][\text 1,1,2,3,5,8,13,…][1,1,2,3,5,8,13,…]。
现在,请求出。
输入描述:
输入共一行,包含4\text 44个正整数a,b,m,n(1≤a,b,m≤109,1≤n≤105)a,b,m,n(1\le a,b,m\le 10^9,1\le n \le 10^5)a,b,m,n(1≤a,b,m≤109,1≤n≤105)。
输出描述:
输出共一行,包含一个非负整数表示答案。由于结果可能较大,你只需要输出答案对109+7=1,000,000,00710^9+7=1,000,000,007109+7=1,000,000,007取模的结果。
示例
输入
1 1 2 4
输出
8
思路:第一反映想到的就是枚举数列和快速幂,结果交了一发WA了。原来,题目有一个坑,就是幂数和底数的模数是不一样的,需要使用欧拉降幂公式。
AC的代码如下:
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
typedef long long ll;
const int N = 100010;
const ll mod = 1e9 + 7;
ll s[N];
ll qmi(ll a, ll k, ll p) //快速幂
{
ll res = 1;
while(k)
{
if(k & 1) res = res * a % p;
k >>= 1;
a = a * a % p;
}
return res;
}
int main()
{
ll n, a, b, m;
cin >> a >> b >> m >> n;
s[1] = 1, s[2] = 1;
for(int i = 3; i <= n; i++)
s[i] = (a * s[i - 1] + b * s[i - 2]) % (mod - 1); //欧拉降幂
cout << qmi(m, s[n], mod) % mod << endl;
}