动态规划(矩阵快速幂优化DP,week14作业)

D - Q老师染砖(选做)

题目

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

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

为了使工作效率更高,Q老师 想要知道一共有多少种方案可以使染色效果最佳,你能帮帮他吗?
第一行为 T,代表数据组数。(1 ≤ T ≤ 100)

接下来 T 行每行包括一个数字 N,代表有 N 块砖。(1 ≤ N ≤ 1e9)
输出满足条件的方案数,答案模 10007

2
1
2

2
6

思路

矩阵快速幂:
设A为n*n方阵,k为一个常数,若求A^k时间超限时,可以借鉴快速幂的方法进行求解,复杂度由O(N0缩短为O(logN)。
快速幂具体为:
在这里插入图片描述而将线性递推式(这里以斐波那契为例)转化为矩阵快速幂求解的方法如下图所述:
在这里插入图片描述本题解析:
如下图所示,列出状态方程后,转化为矩阵快速幂,然后求解。
在这里插入图片描述
在这里插入图片描述

代码

#include<iostream>
#include<algorithm>
#include<memory.h>
using namespace std;

const int maxn=1e9+50;
const int N=3;

struct matrix
{
	int x[N][N];
	matrix operator*(const matrix& t)
	{
		matrix res;
		for(int i=0;i<N;i++)
			for(int j=0;j<N;j++)
			{
				res.x[i][j]=0;
				for(int k=0;k<N;k++)
					res.x[i][j]+=x[i][k]*t.x[k][j]%10007;
				res.x[i][j]%=10007;
			}
		return res;
	}
	matrix()
	{
		memset(x,0,sizeof x);
	}
	matrix(const matrix& t)
	{
		memcpy(x,t.x,sizeof x);
	}
};
matrix quick_pow(matrix a,int k)
{
	matrix res;
	for(int i=0;i<N;i++)res.x[i][i]=1;
	while(k)
	{
		if(k&1)res=res*a;
		a=a*a;
		k>>=1;
	}
	return res;
}


void solve(int n)
{
	/*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];*/
	matrix a;
	a.x[0][0]=2,a.x[0][1]=0,a.x[0][2]=1;
	a.x[1][0]=0,a.x[1][1]=2,a.x[1][2]=1;
	a.x[2][0]=2,a.x[2][1]=2,a.x[2][2]=2;
	a=quick_pow(a,n-1);
	int ans=0;
	ans=a.x[0][0]*2+a.x[0][1]*0+a.x[0][2]*2;
	ans%=10007;
	cout<<ans<<endl;
	
}
int main()
{
	int t;
	cin>>t;
	while(t--)
	{
		int n;
		cin>>n;
		solve(n);
		
	}
	return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值