-
Notifications
You must be signed in to change notification settings - Fork 13
/
Nth_Fibonacci.cpp
53 lines (46 loc) · 1.25 KB
/
Nth_Fibonacci.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
//
void fib(ll n, ll&x, ll&y){
if(n==0){
x = 0;
y = 1;
return ;
}
if(n&1){
fib(n-1, y, x);
y=(y+x)%mod;
}else{
ll a, b;
fib(n>>1, a, b);
y = (a*a+b*b)%mod;
x = (a*b + a*(b-a+mod))%mod;
}
}
//
long long fib(int n)
{
if(n==0) return 0;
if(n==1 || n==2) return 1;
if(n==3) return 2;
unsigned long long res[2][2]={{1,1},{1,0}};
unsigned long long mat[2][2]={{1,1},{1,0}};
n-=2;
long long p,q,r,s;
while(n)
{
if(n&1)
{
p=(res[0][0]*mat[0][0] + res[0][1]*mat[1][0])%MOD;
q=(res[0][0]*mat[0][1] + res[0][1]*mat[1][1])%MOD;
r=(res[1][0]*mat[0][0] + res[1][1]*mat[1][0])%MOD;
s=(res[1][0]*mat[0][1] + res[1][1]*mat[1][1])%MOD;
res[0][0]=p;res[0][1]=q;res[1][0]=r;res[1][1]=s;
}
n>>=1;
p=(mat[0][0]*mat[0][0] + mat[0][1]*mat[1][0])%MOD;
q=(mat[0][0]*mat[0][1] + mat[0][1]*mat[1][1])%MOD;
r=(mat[1][0]*mat[0][0] + mat[1][1]*mat[1][0])%MOD;
s=(mat[1][0]*mat[0][1] + mat[1][1]*mat[1][1])%MOD;
mat[0][0]=p;mat[0][1]=q;mat[1][0]=r;mat[1][1]=s;
}
return res[0][0];
}