FZU 1683 纪念SlingShot

Problem 1683 纪念SlingShot

Accept: 525    Submit: 1780
Time Limit: 1000 mSec    Memory Limit : 32768 KB

Problem Description

已知 F（n）=3 * F（n-1）+2 * F（n-2）+7 * F（n-3），n>=3，其中F（0）=1，F（1）=3，F（2）=5，对于给定的每个n，输出F（0）+ F（1）+ …… + F（n） mod 2009。

Input

第一行是一整数m，代表总共有m个cases。

Output

对于每个case，输出一行。格式见样例，冒号后有一个空格。

Sample Input

236

Sample Output

Case 1: 37Case 2: 313

Source

FOJ月赛-2009年2月- Coral

这道题目只需要构造矩阵,比较简单,但需要注意的一点是

直接矩阵快速幂会超时,需要预处理出2^i次的矩阵(因为每次模的是固定的)

#include<bits/stdc++.h>
using namespace std;
const int MOD=2009;
struct Matrix{
    int m[4][4];
    void init(){
        memset(m,0,sizeof(m));
    }
    Matrix operator *(const Matrix &b) const{
        Matrix ret;
        ret.init();
        for(int i=0;i<4;i++)
            for(int j=0;j<4;j++){
                if(m[i][j]!=0){
                    for(int k=0;k<4;k++)
                        ret.m[i][k]=(ret.m[i][k]+m[i][j]*b.m[j][k])%MOD;
                }
            }
        return ret;
    }
};

Matrix mat[35];

Matrix Pow(int n){
    Matrix ans;
    ans.init();
    ans.m[0][0]=4,ans.m[1][0]=5;
    ans.m[2][0]=3,ans.m[3][0]=1;
    int cnt=0;
    while(n){
        if(n&1){
            ans=mat[cnt]*ans;   //不能颠倒顺序,因为乘法不满足交换律
        }
        n>>=1;
        cnt++;
    }
    return ans;
}

int main(){
    Matrix base;
    base.init();
    base.m[0][0]=1,base.m[0][1]=1,base.m[0][2]=0,base.m[0][3]=0;
    base.m[1][0]=0,base.m[1][1]=3,base.m[1][2]=2,base.m[1][3]=7;
    base.m[2][0]=0,base.m[2][1]=1,base.m[2][2]=0,base.m[2][3]=0;
    base.m[3][0]=0,base.m[3][1]=0,base.m[3][2]=1,base.m[3][3]=0;
    mat[0]=base;
    for(int i=1;i<=32;i++)
        mat[i]=mat[i-1]*mat[i-1];
    int _,n;
    scanf("%d",&_);
    for(int t=1;t<=_;t++){
        printf("Case %d: ",t);
        scanf("%d",&n);
        if(n==0){
            printf("1\n");
            continue;
        }
        printf("%d\n",Pow(n-1).m[0][0]);
    }
    return 0;
}