HDU 1005 Number Sequence

题目链接：Number Sequence

解题思路：矩阵幂就可以实现，注意n为1或者2的时候要特判一下。

#include<stdio.h>
#include<string.h>

__int64 ans[2][2], tem[2][2];

void mutl(__int64 a[2][2],__int64 b[2][2]){
    __int64 c[2][2]; 
    int i, j, k;
    memset(c, 0, sizeof(c));
    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] + c[i][j]) % 7) + 7) % 7;
            } 
        }
    }
    for(i = 0; i < 2; i++){
        for(j = 0; j < 2; j++){
            a[i][j] = c[i][j];
        }
    }
}

int main(){
    int a, b, n;
    while(scanf("%d%d%d", &a, &b, &n) && (a || b || n)){
        if(n == 1 || n == 2){
            printf("1\n");
            continue;
        }
        n -= 2;
        memset(ans, 0, sizeof(ans));
        ans[0][0] = ans[1][1] = tem[1][0] = 1;
        tem[0][0] = a;
        tem[0][1] = b;
        tem[1][1] = 0;
        while(n){
            if(n & 1){
                mutl(ans, tem);    
            }
            n >>= 1;
            mutl(tem, tem);
        }
        printf("%I64d\n", (((ans[0][0] + ans[0][1]) % 7) + 7) % 7);
    }
    return 0;
}