HDU2604 Queuing

　　原题链接：http://acm.hdu.edu.cn/showproblem.php?pid=2604

　　矩阵乘法，得到递推公式，构造系数矩阵，然后进行快速幂。

　　

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

int L, M;

struct mat
{
    int m[4][4];
    mat()
    {memset(m, 0, sizeof m);}
    
    void init()
    {
        memset(m, 0, sizeof m);
        m[0][0] = m[0][2] = m[0][3] = m[1][0] = m[2][1] = m[3][2] = 1;
    }
    
    void unit()
    {
        for(int i = 0; i < 4; i ++)
        for(int j = 0; j < 4; j ++)
            m[i][j] = (i == j);
    }
}A, B;


int F4[5] = {0, 2, 4, 6, 9};

mat operator *(mat a, mat b)
{
    mat c;
    for(int i = 0; i < 4; i ++)
    for(int j = 0; j < 4; j ++)
    for(int k = 0; k < 4; k ++)
        c.m[i][j] = (c.m[i][j] + a.m[i][k] * b.m[k][j]) % M;
    
    return c;
}


int main()
{
    while(scanf("%d%d", &L, &M) == 2)
    {
        if(L <= 4)
        {
            printf("%d\n", F4[L] % M);
            continue;
        }
        A.init(); B.unit();
        for(L -= 4; L; L >>= 1, A = A * A)
            if(L & 1) B = B * A;
        int ans = 0;
        for(int i = 0; i < 4; i ++)
            ans += B.m[0][i] * F4[4-i];
        printf("%d\n", ans % M);
    }
    return 0;
}