HDU 6030 DP + 矩阵快速幂

题意：

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=6030
有红蓝两种颜色的珠子构成一个串（不是环），需要满足在任意长度为素数的区间内，红色珠子数目都要不小于蓝色珠子数目，问长度为n的情况下一共有多少种可能组成的项链。

---

思路：

递推。dp[i][0]表示长度为i且末尾为蓝色珠子的方案数，dp[i][1]表示长度为i且末尾为红色珠子的方案数，可以得到转移方程：
dp[i][1] = dp[i-1][0] + dp[i-1][1]
dp[i][0] = dp[i-2][1]
合并一下两个方程，令f[i] = dp[i][0] + dp[i][1]，可以得到最终的方程：
f[i] = f[i-1] + f[i-3]
接下来快速幂即可。
需要注意的是f[1]=2。因为只有一个的珠子的情况下没有素数区间。

---

代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const LL MOD = 1e9 + 7;

struct mat{
    LL p[5][5], sz;
    mat operator + (const mat &x) {
        mat tp; tp.sz = sz;
        for (int i = 1; i <= sz; i++)
            for (int j = 1; j <= sz; j++)
                tp.p[i][j] = (p[i][j] + x.p[i][j]) % MOD;
        return tp;
    }
    mat operator * (const mat &x) {
        mat tp; tp.sz = sz;
        for (int i = 1; i <= sz; i++)
            for (int j = 1; j <= sz; j++) {
                tp.p[i][j] = 0;
                for (int k = 1; k <= sz; k++)
                    tp.p[i][j] = (tp.p[i][j] + p[i][k] * x.p[k][j] % MOD) % MOD;
            }
        return tp;
    }
}unit, Am;

mat pow_mod(mat a, LL n) {
    mat res = unit;
    while (n) {
        if (n & 1) res = res * a;
        a = a * a;
        n >>= 1;
    }
    return res;
}

void init() {
    memset(&unit, 0, sizeof(unit));
    unit.sz = 3;
    for (int i = 1; i <= unit.sz; i++) unit.p[i][i] = 1;
}

int main() {
    int T;
    scanf("%d", &T);
    while (T--) {
        LL n;
        scanf("%I64d", &n);
        if (n == 2) {
            puts("3");
            continue;
        }
        init();
        memset(&Am, 0, sizeof(Am));
        Am.sz = 3;
        Am.p[1][1] = 1; Am.p[1][3] = 1;
        Am.p[2][1] = 1;
        Am.p[3][2] = 1;
        Am = pow_mod(Am, n - 2);
        LL ans = (((3 * Am.p[1][1]) % MOD + 2 * Am.p[1][2] % MOD) % MOD + Am.p[1][3]) % MOD;
        printf("%I64d\n", ans);
    }
    return 0;
}