Q老师染砖

题意:

衣食无忧的 Q老师 有一天突发奇想,想要去感受一下劳动人民的艰苦生活。

具体工作是这样的,有 N 块砖排成一排染色,每一块砖需要涂上红、蓝、绿、黄这 4 种颜色中的其中 1 种。且当这 N 块砖中红色和绿色的块数均为偶数时,染色效果最佳。

为了使工作效率更高,Q老师 想要知道一共有多少种方案可以使染色效果最佳,你能帮帮他吗?

Input
第一行为 T,代表数据组数。(1 ≤ T ≤ 100)

接下来 T 行每行包括一个数字 N,代表有 N 块砖。(1 ≤ N ≤ 1e9)

Output
输出满足条件的方案数,答案模 10007。

Sample Input

2
1
2

Sample Output

2
6

思路:

定义状态A[i],B[i],C[i]分别为前i块砖的情况下,红绿块数全为偶数、全为奇数、一奇一偶的方案数。
不难看出A[1]=2,B[1]=0,C[1]=2

转移过程:

A[i]=2×A[i−1]+C[i−1] 
B[i]=2×B[i−1]+C[i−1]
C[i]=2×A[i−1]+2×B[i−1]+2×C[i−1]

​由于N的范围很大,O(n)的复杂度是无法接受的,可以采用矩阵快速幂进行优化。
矩阵递推式如下所示:
在这里插入图片描述
基于矩阵乘法的结合律,累乘可得:
在这里插入图片描述
利用矩阵快速幂求得常数矩阵的n-1次方,从而便可得到答案。
在这里插入图片描述
代码有注释。

代码:

#include <algorithm>
#include <cstdio>
#include <cstring>
using namespace std;
const int maxn = 1e9+5;
const int N=4;
int n,mod= 10007;

struct Matrix
{
    int m[N][N];

    void ini()
    {//初始化A
        memset(m, 0, sizeof(m));
        m[1][3]=m[2][3]=1;
        m[1][1]=m[2][2]=m[3][1]=m[3][2]=m[3][3]=2;
    }
    void ini1()
    {//初始化F(1)
        memset(m, 0, sizeof(m));
        m[1][1]=2;
        m[2][1]=0;
        m[3][1]=2;
    }
    Matrix() {}

    Matrix(const Matrix &t)
    {
        memcpy(m, t.m, sizeof(m));
    }

    Matrix operator*(const Matrix &t) const
    {//乘法
        Matrix res;
        for (int i = 1; i < N ; i++)
        {
            for (int j = 1; j < N; j++)
            {
                res.m[i][j] = 0;
                for (int k = 1; k < N; k++)
                {
                    res.m[i][j] += (m[i][k] * t.m[k][j])%mod;
                }
                res.m[i][j]%=mod;  //注意取模
            }
        }
        return res;
    }
};
Matrix quick_pow(Matrix a, Matrix b, int x)
{  //快速幂
    while (x)
    {
        if (x & 1)
        {
            b = a * b;
        }  
        a = a * a;
        x >>= 1;
    }
    return b;
}

Matrix A, f;
int main()
{
    int T;scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        A.ini();
        f.ini1();
        f=quick_pow(A,f,n-1);
        printf("%d\n",f.m[1][1]);
    }


    return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值