zoj 3690 DP+矩阵快速幂

题意：N个人站成一行，有M个数可供任意选择。唯一要求是，当相邻两人取相同数时，这个数必须大于K。

dp[i][0] 表示第i个人取[1..k]的时候的排列种数，dp[i][1] 表示第i个人取[k+1..m]时候的排列种数，得到方程：

dp[i][0]  =  dp[i-1][1]*k         +    dp[i-1][0]*(k-1)

dp[i][1]  =  dp[i-1][1]*(m-k)  +     dp[i-1][0]*(m-k)

初始dp[1][0]=k dp[1][1]=m-k  每次i+1时 都要按上述运算，所以可以得到矩阵

再根据矩阵快速幂，能够把O(n)的算矩阵幂用O(logn)算出

while(N){
                 if(N&1)
                        res=res*A;
                 N>>=1;
                 A=A*A;
}

N是次幂 里面的乘号，是矩阵乘的运算，res是结果矩阵。

第3行代码每进行一次，二进制数就少了最后面的一个1。二进制数有多少个1就第3行代码就执行多少次。

详细请戳这里 http://www.cnblogs.com/yan-boy/archive/2012/11/29/2795294.html

这样把最后结果dp[n][0]+dp[n][1]输出就行了

#include<stdio.h>
#include<string.h>
#include<algorithm>
#define mod 1000000007
using namespace std;
struct P{
	long long  matrix[2][2];
}it;
long long  n,k,m;
P mup(P a,P b){
	P now;
	now.matrix[0][0]=((a.matrix[0][0]*b.matrix[0][0])%mod+(a.matrix[0][1]*b.matrix[1][0])%mod)%mod;
	now.matrix[0][1]=((a.matrix[0][0]*b.matrix[0][1])%mod+(a.matrix[0][1]*b.matrix[1][1])%mod)%mod;
	now.matrix[1][0]=((a.matrix[1][0]*b.matrix[0][0])%mod+(a.matrix[1][1]*b.matrix[1][0])%mod)%mod;
	now.matrix[1][1]=((a.matrix[1][0]*b.matrix[0][1])%mod+(a.matrix[1][1]*b.matrix[1][1])%mod)%mod;
	return now;
}
P find(long long numb){
	P a;
	bool flag=0;
	while(numb){
		if(numb&1){
			if(flag==0){
				a=it;
				flag=1;
			}
			else
				a=mup(a,it);
		//	printf("! numb=%d\n",numb);printf("!! 00=%d 01=%d 10=%d 11=%d\n",a.matrix[0][0],a.matrix[0][1],a.matrix[1][0],a.matrix[1][1]);
		}
		numb=numb>>1;
		it=mup(it,it);
	}
	return a;
}
int main(){
	while(scanf("%lld%lld%lld",&n,&m,&k)!=EOF){
		it.matrix[0][0]=(k-1)%mod;
		it.matrix[0][1]=k%mod;
		it.matrix[1][0]=(m-k)%mod;
		it.matrix[1][1]=(m-k)%mod;
		P a=find(n-1); //printf("!! 00=%d 01=%d 10=%d 11=%d\n",a.matrix[0][0],a.matrix[0][1],a.matrix[1][0],a.matrix[1][1]);
		long long ans1=((a.matrix[0][0]*k)%mod+(a.matrix[0][1]*(m-k))%mod)%mod;
		long long ans2=((a.matrix[1][0]*k)%mod+(a.matrix[1][1]*(m-k))%mod)%mod;
		printf("%lld\n",(ans1+ans2)%mod);
	}
	return 0;
}