hdu 1588

http://acm.hdu.edu.cn/showproblem.php?pid=1588
题目大意:求下标为g(x)=ki+b,(i=0,1,…,n-1)的Fibonacci数列F( g(i) )的和mod m,即求Fib(k0+b)+Fib(k1+b)+Fib(k2+b)+……+Fib(k*(n-1)+b) mod m
整理公式后,和poj 3233相似。

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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
#include<stdio.h>
#include<cstring>
#include<algorithm>
#include<iostream>
#define mem(x,y) memset(x,y,sizeof(x))
#define inf 0x3f3f3f3f
using namespace std;
#define siz 2
#define Mytype long long
int n;
Mytype mod;
struct matrix
{
Mytype a[siz][siz];
matrix operator*(const matrix &y)const
{
matrix res;
mem(res.a,0);
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
if(a[i][j])
for(int k=0;k<n;k++)
res.a[i][k]+=a[i][j]*y.a[j][k],res.a[i][k]%=mod;
return res;
}
matrix operator+(const matrix &y)const
{
matrix res;
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
res.a[i][j]=a[i][j]+y.a[i][j],res.a[i][j]%=mod;
return res;
}
};
matrix qmod(matrix a,int k)
{

matrix res;
mem(res.a,0);
for(int i=0;i<n;i++)
res.a[i][i]=1;
while(k)
{
if(k&1)
res=res*a;
a=a*a;
k>>=1;
}
return res;
}
matrix bin(const matrix &a,int k)
{

if(k==1)
return a;
if(k&1)
return qmod(a,k)+bin(a,k-1);
int half=k/2;
matrix A=bin(a,half);
matrix B=qmod(a,half);
return A+A*B;
}
int main()
{

int b,base,m;
n=2;
while(~scanf("%d%d%d%I64d",&base,&b,&m,&mod))
{
matrix a;
a.a[0][0]=1,a.a[0][1]=1,a.a[1][0]=1,a.a[1][1]=0;
matrix x1=qmod(a,b);
a=qmod(a,base);
matrix x2=bin(a,m-1);
x1=x1+x1*x2;
printf("%I64d\n",x1.a[1][0]);
}
}

EOF