HDU6030 Happy Necklace【规律 + 矩阵快速幂】

Happy Necklace

---

题意： 一串由红珠和蓝珠组成的项链，要求其中任意质数个长度的子串中红珠的数量不少于蓝珠。问n个珠子能够组成多少串不相同的项链。

题解： 质素其实只需要考虑2和3即可（我一开始忽略了3）。只要确定长度2和3的组合数量，往后应该是有规律可循的。

找规律： 假定红珠为1，蓝珠为0

• 长度n = 2 时：

  • 0 1
  • 1 0
  • 1 1

• 长度n = 3 时：从长度为2的基础上，所有项链后加一都成立，唯有末尾有连续1的才能加0。

  • 0 1 1
  • 1 0 1
  • 1 1 1
  • 1 1 0

• 长度n = 4 时：同理只有长度为3的项链中末尾有连续1的才能加0.

  • 0 1 1 1
  • 0 1 1 0
  • 1 0 1 1
  • 1 1 1 1
  • 1 1 1 0
  • 1 1 0 1

• 长度n = 5 时：

  • 0 1 1 1 1
  • 0 1 1 1 0
  • 0 1 1 0 1
  • 1 0 1 1 1
  • 1 0 1 1 0
  • 1 1 1 1 1
  • 1 1 1 1 0
  • 1 1 1 0 1
  • 1 1 0 1 1

  太长便不再写了，但是仔细观察已经可以发现端倪了。假定f(n)表示长度为n的项链的组合数量，长度为n的情况可以直接在n - 1的项链末加1得到，所以f(n) = f(n - 1) + ?而这个问号，则应该是长度n - 1的项链中，末尾有连续1的数量，如果我们去追溯的话，这个数量等于f(n - 3)。最终得到递推公式为为：f(n) = f(n - 1) + f(n - 3)

矩阵快速幂： 然后回头看一眼数据范围：1e18。可真有你的啊，这么大的范围，常规操作肯定是过不了的。结合这个递推公式，我们应该借用矩阵快速幂来解决。
  在得到递推公式之后，我们希望能有两个矩阵，通过矩阵快速幂的方法，去求得1e18范围下的答案。除了要有矩阵快速幂的模板之外，最关键的在于矩阵的构建（我真的太笨了，找了好多博客，消化了好久才有点知道该如何构建）。【所以为什么要学线性代数啊，这不就用到了矩阵 了么】

[f[n - 1], f[n - 2], f[n - 3]] * A = [f[n], f[n - 1], f[n - 2]]
$$A=\left[\begin{array}{ccc}1& 1& 0\\ 0& 0& 1\\ 1& 0& 0\end{array}\right]$$
然后就可以起飞了，芜湖，【不是。

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

ll _T, n;
struct mat{ ll mat[3][3];};
mat operator *(mat &a, mat &b) {
	mat c;
    memset(c.mat, 0, sizeof c.mat);
    for(int k = 0; k < 3; ++k)
        for(int i = 0; i < 3; ++i)
            for(int j = 0; j < 3; ++j)
                c.mat[i][j] = (c.mat[i][j] + a.mat[i][k] * b.mat[k][j] % mod) % mod;
    return c;
}
mat power(mat a, ll n) {
	mat c;
	memset(c.mat, 0, sizeof c.mat);	//千万记得初始化
	for(int i = 0; i < 3; i++)	c.mat[i][i] = 1;
	while(n) {
		if(n & 1) c = c * a;
		n >>= 1;
		a = a * a;
	}
	return c;
}
void run() {
	scanf("%lld", &n);
	mat a = {1, 1, 0, 0, 0, 1, 1, 0, 0};
	mat b = {3, 2, 1, 0, 0, 0, 0, 0, 0};
	a = power(a, n - 2);
	a = b * a;
	printf("%lld\n", a.mat[0][0] % mod);
}
int main() {
	scanf("%lld", &_T);
	while(_T--)	run();
    return 0;
}