矩阵快速幂的练习题,需要将题目转化一下才能用矩阵快速幂。
题目大意:有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;
}