ZJNU 1265 推公式

幸运的硬币

Time Limit: 1000MS		Memory Limit: 1000K

Description

众所周知，一个硬币有两面，一面朝上，一面朝下。如果两个硬币同时朝上，或者同时朝下，我们说两个硬币的状态是相同的。如果现在有N个硬币排成一排，显然有2^N种不同的排法。如果存在超过两个连续硬币的状态相同，我们就说这个硬币序列是幸运的。那么这种幸运的硬币序列有多少种呢？

Input

第一行一个整数N。（1<=N<=10^9）表示硬币个数。

Output

一个整数，表示幸运硬币序列的种数，数值可能会非常大，将结果Mod 10007。

Sample Input

34

Sample Output

26

题解：

0 (n<=2) 
2^n - 2 * fib(n-2) - 4 * fib(n-1) (n>=3)

这里fib(1)=1,fib(2)=1,fib(3)=2,fib(4)=3,...(斐波那契数列)

证明：
考虑不幸运的序列，一个序列是不幸运的当且仅当它没有连续超过2个的0或1，那么
用f[i]表示长度为i的不幸运的序列个数。

f[1]=2(0或1)
f[2]=4(00/01/10/11)
f[i]=f[i-1](对于任意长度为i-1的不幸运序列{an}，添加一个与a[i-1]颜色不同的硬币) +f[i-2](对于任意长度为i-2的不幸运序列{an}，添加两个与a[i-2]颜色不同的硬币)
(i>=3)

上述转移的可行性在于，任意的不幸运序列，总可以通过去掉尾部的连续的同色硬币，成为一个不幸运序列，并且长度变短。而不幸运序列中连续的同色硬币只有两种情况：1个或2个。

设f[i]=a[i] * f[1]+b[i] * f[2]，则
f[1]=1 * f[1] + 0 * f[2]
f[2]=0 * f[1] + 1 * f[2]
f[3]=1 * f[1] + 1 * f[2]
f[4]=1 * f[1] + 2 * f[2]
f[5]=2 * f[1] + 3 * f[2]
...
不难发现，a[i],b[i]都是下标平移后的斐波那契数列。于是有上述公式。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
struct matrix{
	int a[3][3];
};
matrix multi(matrix a,matrix b){
	matrix ans;
	memset(ans.a,0,sizeof(ans.a));
	int i,j,k;
	for(i=1;i<=2;i++){
		for(j=1;j<=2;j++){
			if(!a.a[i][j])continue;
			for(k=1;k<=2;k++){
				ans.a[i][k]=(ans.a[i][k]+a.a[i][j]*b.a[j][k])%10007;
			}
		}
	}
	return ans;
}
matrix quick(matrix a,int k){
	matrix ans;
	ans.a[1][1]=ans.a[2][2]=1;
	ans.a[1][2]=ans.a[2][1]=0;
	while(k){
		if(k&1){
			ans=multi(ans,a);
		}
		k/=2;
		a=multi(a,a);
	}
	return ans;
}
int pow(int a,int k){
	int ans=1;
	while(k){
		if(k&1){
			ans=ans*a%10007;
		}
		k>>=1;
		a=a*a%10007;
	}
	return ans;
}
int main(){
	int n;
	scanf("%d",&n);
	if(n<=2){
		printf("0\n");
		return 0;
	}
	matrix t;
	t.a[1][1]=t.a[1][2]=t.a[2][1]=1;
	t.a[2][2]=0;
	t=quick(t,n-2);
	int ans=pow(2,n)-4*t.a[1][1]-2*t.a[2][1];
	while(ans<0)ans+=10007;
	printf("%d\n",ans);
	return 0;
}