HDU6030

dp+矩阵快速幂
dp[i][1]表示一共有i个珠子,1表示以红色珠子结尾;dp[i][0],0表示以蓝色珠子结尾。
dp[i][1]=dp[i-1][0]+dp[i-1][1]
dp[i][0]=dp[i-2][1]

代码:

import java.util.*;
public class Main {
	static long N=1000000007;
	public static void main(String[] args) {
		// TODO 自动生成的方法存根
		Scanner sc=new Scanner(System.in);
		int n=sc.nextInt();
		for(int i=0;i<n;i++) {
			long x=sc.nextLong();
			System.out.println(fastMul(x));
		}
	}
	static long[][] mul(long[][] a,long[][] b) {
		long[][] res=new long[a.length][b[0].length];
		for(int i=0;i<a.length;i++)
			for(int j=0;j<b[0].length;j++)
				for(int k=0;k<b.length;k++)
					res[i][j]=(res[i][j]+a[i][k]*b[k][j]%N)%N;
		return res;
	}
	static long fastMul(long n) {
		if(n==2)
			return 3;
		long[][] ans=new long[3][3];
		long[][] base=new long[3][3];
		ans[0][0]=ans[1][1]=ans[2][2]=1;
		base[0][0]=base[0][1]=base[1][2]=base[2][0]=1;
		n=n-2;
		while(n>0) {
			if(n%2==1)
				ans=mul(ans,base);
			base=mul(base,base);
			n>>=1;
		}
		long[][] a=new long[3][1];
		a[0][0]=2;
		a[1][0]=1;
		a[2][0]=1;
		ans=mul(ans,a);
		return (ans[0][0]+ans[1][0])%N;
	}
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值