和我以前写的斐波拉契数列一样的,就是改变下数组就好
#include <stdio.h>
#include <stdlib.h>
void multiply(int a[][3],int b[][3])
{
int i ,j,k;
int c[3][3]={{0}};
for(i=0;i<2;i++)
for(j=0;j<2;j++)
for(k=0;k<2;k++)
c[i][j]+=a[i][k]*b[k][j]%7;
for(i=0;i<2;i++)
for(j=0;j<2;j++)
a[i][j]=c[i][j]%7;
}
void powr(int a[][3],int n)
{
int i ,j;
int b[3][3];
for(i=0;i<2;i++)
for(j=0;j<2;j++)
b[i][j]=a[i][j];
if(n>1)
{
if(!(n%2))
{
powr(a,n/2);
multiply(a,a);
}
else
{
powr(a,(n-1)/2);
multiply(a,a);
multiply(a,b);
}
}
}
int main()
{
int a,b,n;
int matrix[3][3]={{0}};
while(scanf("%d %d %d",&a,&b,&n)&&(a||b||n))
{
matrix[0][0]=a;
matrix[0][1]=b;
matrix[1][0]=1;
matrix[1][1]=0;
if(n<3)
printf("1\n");
else
{
powr(matrix,n-2);
printf("%d\n",(matrix[0][0]+matrix[0][1])%7);
}
}
return 0;
}