ZOJ 3690 Choosing number

#include <cstdio>
#include <cstring>
#include <cmath>
#include <vector>
#include <iostream>
#include <algorithm>

using namespace std;

#define LL long long
#define mod 1000000007

struct matrix{
    LL f[5][5];
};

matrix mul(matrix a,matrix b){
	int i,j,k;
	matrix ans;
	memset(ans.f,0,sizeof(ans.f));
 	for(i=1;i<=2;i++){
 		
        for(j=1;j<=2;j++)
            for(k=1;k<=2;k++)
                ans.f[i][j]=(ans.f[i][j]+a.f[i][k]*b.f[k][j])%mod;
    }
	return ans;
}
matrix pow(matrix e,LL n){
    matrix s;
    s.f[1][1]=s.f[2][2]=1;
    s.f[1][2]=s.f[2][1]=0;
	while(n){
		if(n & 1)
			s=mul(s,e);
		e=mul(e,e);
		n >>=1;
	}
	return s;
}
int main(){
	matrix e;
	LL m,n,k;
	while(scanf("%lld%lld%lld",&m,&n,&k)!=EOF){
		e.f[1][1]=n-k;e.f[1][2]=k;
		e.f[2][1]=n-k;e.f[2][2]=k-1;
		e=pow(e,n-1);
		//printf("%lld %lld %lld %lld\n",e.f[1][1],e.f[1][2],e.f[2][1],e.f[2][2]);
		printf("%lld\n",((m-k)*e.f[1][1]+(m-k)*e.f[1][2]+k*e.f[2][1]+k*e.f[2][2])%mod);
	}
}

思路看的还是大神的思路，哎，太菜了

下面说明下思路：
    
    f[i]，表示前i-1个人的选择，且第i个人选择大于k的值
    g[i], 表示前i-1个人的选择，且第i个人选择小于k的值
    可得到递推式
    f[i]=(m-k)*f[i-1]+(m-k)*g[i-1]
    g[i]=k*f[i-1]+(k-1)*g[i-1]
    可得到矩阵：
    |f[i] g[i]|=|f[i-1] g[i-1]|*|m-k k  |
                                |m-k k-1|
                                
       |f[n] g[n]|=|f[1] g[1]|* |m-k k  |^(n-1)
                                |m-k k-1|
    ans=f[n]+g[n];