HDU1005 Number Sequence

#include<stdio.h>
void fast_pow(int a[][2], int b[][2]) 
{
  int c[2][2];

  c[0][0] = (a[0][0]*b[0][0] + a[0][1] * b[1][0]) % 7;
  c[0][1] = (a[0][0]*b[0][1] + a[0][1] * b[1][1]) % 7;
  c[1][0] = (a[1][0]*b[0][0] + a[1][1] * b[1][0]) % 7;
  c[1][1] = (a[1][0]*b[0][1] + a[1][1] * b[1][1]) % 7;

  a[0][0]=c[0][0];
  a[0][1]=c[0][1];
  a[1][0]=c[1][0];
  a[1][1]=c[1][1];
}
int main() 
{
  int A,B,n;
  int a[2][2],b[2][2];
  while (scanf("%d%d%d", &A,&B,&n) , n || A || B) {
      a[0][0]=A;
      a[0][1]=B;
      a[1][0]=b[0][0]=b[1][1]=1;
      a[1][1]=b[0][1]=b[1][0]=0;
      n--;
      while (n){
          if (n & 1)
              fast_pow(b, a);
          fast_pow(a, a);
          n >>= 1;
      }
      printf ("%d\n", (b[1][0]+b[1][1])%7);//因为忘了对7求余，贡献好几个WA 
  }
  return 0;
}

快速幂的典型例题，本题还可以利用求循环阶来求！代码如下：

/*
循环阶最大为49，因为每一个f[i]都有它前面的两个f[i-1]和f[i-2]决定，
而每个f[i-1]和f[i-2]的取值范围都是0~6共七个，所以f[i]的变换范围是7*7=49个，
所以循环阶一定不超过49. 
*/ 
#include <stdio.h>
#include <string.h>
#include <math.h>
int f[50];
int main()
{
    int a,b,n,i,j;
    int k[7][7];
    f[1]=1;f[2]=1;
    while (scanf("%d%d%d",&a,&b,&n),a||b||n)
    {
        memset(k,0,sizeof(k));
        k[1][1]=2;
        for (i=3;i<=50;i++)
        {
            f[i]=a*f[i-1]+b*f[i-2];
            f[i]%=7;
            if (k[f[i]][f[i-1]]) break;
            k[f[i]][f[i-1]]=i;
        }
        j=i-k[f[i]][f[i-1]];//计算机循环阶 
        n-=k[f[i]][f[i-1]];//先减去循环阶前的乱序部分 
        n%=j;
        if (n==0) n+=j;//如果恰好是0，需加1 
        n+=k[f[i]][f[i-1]];//最后再加上刚才减去的部分 
        printf("%d\n",f[n]);
    }
}