HDU-6030-Happy Necklace

HDU-6030-Happy Necklace

传送门

这道题是矩阵快速幂~

题目大意：小Q想为自己的女朋友准备一条项链，他有红色的珠子和蓝色的珠子（在这里我想说哪里有女生喜欢红色+蓝色珠子组成的项链emmmm），然后他的女朋友的喜欢条件是：任取这条项链的长度为素数的子串，使得红色的珠子不少于蓝色的珠子。
问小Q有多少种组合珠子的方式使得项链可以让他的女朋友喜欢。
其中这条项链是一条单链，不是环。

我们根据题目要求。
f(n)代表n个珠子有多少种不同的满足要求的组合。
假设a代表红色的珠子，b代表蓝色的珠子。

当n = 2时：f(2) = 3;
方案：aa ab ba
当n = 3时：f(3) = 4;
方案：aaa aab aba baa
当n = 4时：f(4) = 6;
方案：aaaa aaba abaa abab baab baaa
当n = 5时：f(5) = 9;
方案：aaaaa aaaab aabaa abaaa abaab ababa baaba baaab baaaa

以上是我手动操作出来的: )

我们可以发现：最小的素数是2，我们只需要满足任意选出来的素数长度为2，3的子串满足红色数目不少于蓝色数目就行。
在上一种情况下如果末尾以b（蓝色）结尾。那么下一个一定是a。
或者说我们在b的末尾加上aab可以满足要求。

所以根据找规律我们可以发现：
我们的递推式为：f(n) = f(n - 1) + f(n - 3);
这样下来我们可以假设f(1) = 2;方案为：a b（题目要求是n从2开始的）

然而题目数据涉及到1e12;
普通递推肯定不行。这里用矩阵快速幂来解决问题~

综上分析，我们就可以直接套用模板了~
注意取模和ll类型~

代码部分：

#include <bits/stdc++.h>
#define mst(a, n) memset(a, n, sizeof(a))
using namespace std;
typedef long long ll;
const int mod = 1e9 + 7;

ll n;
struct Matrix
{
	ll m[3][3];
	Matrix()
	{
		mst(m, 0);
	}
};

Matrix Multi(Matrix a, Matrix b)
{
	Matrix res;
	for (int i = 0; i < 3; i++)
	{
		for (int j = 0; j < 3; j++)
		{
			for (int k = 0; k < 3; k++)
			{
				res.m[i][j] = (res.m[i][j] + a.m[i][k] * b.m[k][j]) % mod;
			}
		}
	}
	return res;
}

ll npow(Matrix a, ll n)
{
	Matrix res;
	for (int i = 0; i < 3; i++)
	{
		res.m[i][i] = 1;
	}
	while (n)
	{
		if (n & 1)
		{
			res = Multi(res, a);
		}
		a = Multi(a, a);
		n >>= 1;
	}
	return (res.m[0][0] * 4 + res.m[0][1] * 3 + res.m[0][2] * 2) % mod;
}

int main()
{
	int t;
	cin >> t;
	while (t--)
	{
		scanf ("%I64d", &n);
		if (n == 2)
		{
			cout << "3\n";
			continue;
		}
		Matrix res;
		res.m[0][0] = res.m[0][2] = res.m[1][0] = res.m[2][1] = 1;
		cout << npow(res, n - 3) % mod << endl;
	}
	return 0;
}