51nod 1668 非010串

dp[i]=2*dp[i-1]-dp[i-2]+dp[i-3]

由于有减号，所以矩阵快速幂时，可能会产生负数，要处理掉。

#include<bits/stdc++.h>
using namespace std;

const int siz=3;
long long mod=1e9+7;

struct mtx
{
	long long a[siz][siz];
}unit;

mtx multi(mtx m1,mtx m2)
{
	int i,j,k;
	mtx ret;
	memset(ret.a,0,sizeof(ret.a));
	for(i=0;i<siz;i++)
		for(j=0;j<siz;j++)
			for(k=0;k<siz;k++)
				ret.a[i][j]=(ret.a[i][j]+m1.a[i][k]*m2.a[k][j]%mod+mod)%mod;
	return ret;
}

mtx pow(mtx m,long long p)
{
	mtx ret;
	memset(ret.a,0,sizeof(ret.a));
	for(int i=0;i<siz;i++)
		ret.a[i][i]=1;
	while(p)
	{
		if(p&1)
			ret=multi(ret,m);
		m=multi(m,m);
		p>>=1;
	}
	return ret;
}

long long cal(long long x)
{
	if(x==0)	return 0;
	if(x==1)	return 2;
	if(x==2)	return 4;
	if(x==3)	return 7;
	mtx ret;
	ret.a[0][0]=7;	ret.a[0][1]=0;	ret.a[0][2]=0;
	ret.a[1][0]=4;	ret.a[1][1]=0;	ret.a[1][2]=0;
	ret.a[2][0]=2;	ret.a[2][1]=0;	ret.a[2][2]=0;
	ret=multi(pow(unit,x-3),ret);
	return ret.a[0][0];
}

int main()
{
	long long n;
	unit.a[0][0]=2;	unit.a[0][1]=-1;	unit.a[0][2]=1;
	unit.a[1][0]=1;	unit.a[1][1]=0;	unit.a[1][2]=0;
	unit.a[2][0]=0;	unit.a[2][1]=1;	unit.a[2][2]=0;
	while(~scanf("%lld",&n))
	{
		printf("%lld\n",cal(n));
	}
}