hdu-2276(Kiki & Little Kiki 2)

矩阵快速幂的练习题，需要将题目转化一下才能用矩阵快速幂。

题目大意：有n个灯，分别编号 1 到 n，它们构成一个环，1 的左边是 n，k 的左边是 k-1。时间每过 1 s，如果灯 i 左边的灯是亮的，则灯 i 切换状态（1 表示亮，0 表示暗）。附链接：http://acm.hdu.edu.cn/showproblem.php?pid=2276。

大体思路：这里的难点在于如何想到用快速幂，因为题目所给的秒数很大，单纯模拟的话肯定超时，因此可以用矩阵快速幂来解。不过，构造矩阵也需要一点时间，这也可以算作难点，然后是一个小技巧，就是矩阵相乘后每一位与 2 相余，这一点想到了的话用矩阵快速幂的思路也就清晰了。构造矩阵需要想到灯 i 的状态和灯 i 左边的状态有联系，接着构造一个n×n 方，对角线为 1，代表灯 i 自身，对角线的上方就是灯 i 左边的灯，也置为1，由此，矩阵构造完毕。系数矩阵C如下图所示

1	1
	1	1
		1	1
			1	1
				1	1
					1	1
1						1

而开始的矩阵为题目给的灯初始状态T。则该题目就转为求 T*C^M。

以下为ac代码：

#include<iostream>
#include<cstring>
using namespace std;

const int maxn=105;
const int mod=2;   //这一点很巧妙，没有余数为2，矩阵快速幂也就没法应用在该题上
int n;

struct Matrix{
    long long mat[maxn][maxn];
};
Matrix unit;  //单位矩阵

Matrix mul(Matrix a,Matrix b){
    Matrix c;
    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++){
            c.mat[i][j]=0;
            for(int k=0;k<n;k++){
                c.mat[i][j]+=a.mat[i][k]*b.mat[k][j];
                c.mat[i][j]%=mod;
            }
    }
    return c;
}
Matrix mat_pow(Matrix a,long long p){
    Matrix temp=unit;
    while(p){
        if(p%2)
            temp=mul(temp,a);
        a=mul(a,a);
        p/=2;
    }
    return temp;
}

int main(){
    long long M;
    char s[maxn];
    memset(unit.mat,0,sizeof(unit.mat));
    for(int i=0;i<maxn;i++)    //构造单位矩阵
        unit.mat[i][i]=1;

    while(cin>>M){
        cin>>s;
        n=strlen(s);
        Matrix T,C;
        for(int i=0;i<n;i++)
            T.mat[0][i]=s[i]-'0';  //初始矩阵T
        memset(C.mat,0,sizeof(C.mat));
        for(int i=1;i<n;i++)           //构造系数矩阵
            C.mat[i][i]=C.mat[i-1][i]=1;
        C.mat[0][0]=C.mat[n-1][0]=1;

        Matrix D=mat_pow(C,M);  //系数矩阵快速幂
        D=mul(T,D);             //求最后结果
        for(int i=0;i<n;i++)
            cout<<D.mat[0][i];
        cout<<endl;
    }
    return 0;
}