forked from Hawstein/cracking-the-coding-interview
-
Notifications
You must be signed in to change notification settings - Fork 0
/
8.1.cpp
59 lines (57 loc) · 1.25 KB
/
8.1.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
54
55
56
57
58
59
#include <iostream>
using namespace std;
typedef long long ll;
ll fib(ll n){
if(n < 1) return -1;
if(n == 1 || n == 2) return 1;
else return fib(n-1) + fib(n-2);
}
ll fib1(ll n){
if(n < 1) return -1;
if(n == 1 || n == 2) return 1;
ll a = 1, b = 1;
for(ll i=3; i<=n; ++i){
ll c = a + b;
a = b;
b = c;
}
return b;
}
void mul(ll c[2][2], ll a[2][2], ll b[2][2]){
ll t[4];
t[0] = a[0][0]*b[0][0] + a[0][1]*b[1][0];
t[1] = a[0][0]*b[0][1] + a[0][1]*b[1][1];
t[2] = a[1][0]*b[0][0] + a[1][1]*b[1][0];
t[3] = a[1][0]*b[0][1] + a[1][1]*b[1][1];
c[0][0] = t[0];
c[0][1] = t[1];
c[1][0] = t[2];
c[1][1] = t[3];
}
void pow(ll s[2][2], ll a[2][2], ll n){
while(n > 0){
if(n&1) mul(s, s, a);
mul(a, a, a);
n >>= 1;
}
}
ll fib2(ll n){
if(n < 1) return -1;
if(n == 1 || n == 2) return 1;
ll a[2][2] = { {1, 1}, {1, 0} };
ll s[2][2] = { {1, 0}, {0, 1} };
pow(s, a, n-2);
return s[0][0] + s[0][1];
}
int main(){
for(int i=1; i<20; ++i)
cout<<fib(i)<<" ";
cout<<endl;
for(int i=1; i<20; ++i)
cout<<fib1(i)<<" ";
cout<<endl;
for(int i=1; i<20; ++i)
cout<<fib2(i)<<" ";
cout<<endl;
return 0;
}