HDU1588 Gauss Fibonacci 矩阵应用

/*
由f(n)=f(n-1)+f(n-2)构造矩阵
令
Fn(1,2) = |f(n) f(n-1)|
A(2 2) = |1 1|
|1 0|
F1(1 2) = |1 0|
Fn = F(n-1) * A = F1 * A ^ (n-1)
代入n = g(i) = k * i + b 得：Fn = F1 * A ^ (k * i + b - 1)
1)b > 1
Fn = F1 * A^(b-1) * (A^k)^i
Sum(Fn) = F1 * A^(b-1) * Sum((A^k)^i)
2)b = 1
Fn = F1 * (A^k)^i
Sum(Fn) = F1 * Sum((A^k)^i)
3)b = 0
Fn = F1 * A^(k-1) * (A^k)^i------------------n = n - 1
Sum(Fn) = F1 * A^(k-1) * Sum((A^k)^i)--------n = n - 1
其中Sum((A^k)^i)用solve进行二分求和
*/
#include "Mat.h"
#include <iostream>
using namespace std;

int main()
{
Mat A(2, 2), F1(1, 2), e(2, 2), ans(1, 2), temp(2, 2);
int b, n, k;
while(cin>>k>>b>>n>>mod)
{
//全1矩阵
A.clear(2);A.s[1][1] = 0;
F1.clear(1);
e.clear(1);
ans.clear();

temp = A;
if(b > 1)
{
temp.Er_work(b-1);
F1.Multiply(temp);
}
else if(b == 0)
{
n--;
temp.Er_work(k-1);
F1.Multiply(temp);
}
A.Er_work(k);
A.solve(n-1);
A.Add(e);
F1.Multiply(A);
printf("%I64d\n",F1.s[0][0]);
}
return 0;
}