题意:m表示m秒
下面一行表示n个灯(灯是排成环的,也就是说头尾相接)
灯亮暗由 1 0表示
对于任意一盏灯,当左边灯亮时,下一秒该灯将变换状态
问:输出m秒后灯的状态
因为每盏灯都由前一秒的该灯和该灯左边那盏灯的状态决定,所以可以写出一个矩阵
当有5盏灯时:
1 0 0 0 1
1 1 0 0 0
0 1 1 0 0
0 0 1 1 0
0 0 0 1 1
原来的n乘上上面的矩阵就可以得到下一秒的状态,然后就是矩阵快速幂
#include<stdio.h>
#include<string.h>
#include<math.h>
#include<iostream>
#include<queue>
#define ll int
#define MOD 2
#define Matr 105 //矩阵大小
using namespace std;
inline ll Max(ll a,ll b){return a>b?a:b;}
inline ll Min(ll a,ll b){return a>b?b:a;}
//矩阵快速幂 +++
struct mat//矩阵结构体,a表示内容,size大小
{
int a[105][105],size;
mat()
{
size=0;
memset(a,0,sizeof(a));
}
};
void print(mat m)//输出矩阵信息,debug用
{
int i,j;
printf("%d\n",m.size);
for(i=0;i<m.size;i++)
{
for(j=0;j<m.size;j++)printf("%d ",m.a[i][j]);
printf("\n");
}
}
mat multi(mat m1,mat m2,int mod)//两个相等矩阵的乘法,对于稀疏矩阵,有0处不用运算的优化
{
mat ans=mat(); ans.size=m1.size;
for(int i=1;i<=m1.size;i++)
for(int j=1;j<=m2.size;j++)
if(m1.a[i][j])//稀疏矩阵优化
for(int k=1;k<=m1.size;k++)
ans.a[i][k]=(ans.a[i][k]+m1.a[i][j]*m2.a[j][k])%mod;
return ans;
}
mat quickmulti(mat m,int n,int mod)//二分快速幂
{
mat ans=mat();
int i;
for(i=1;i<=m.size;i++)ans.a[i][i]=1;
ans.size=m.size;
while(n)
{
if(n&1)ans=multi(m,ans,mod);
m=multi(m,m,mod);
n>>=1;
}
return ans;
}
int main(){
int n,i,mul,m[Matr],ans[Matr];
char s[Matr];
while(~scanf("%d",&mul)){
scanf("%s",s);
int n=strlen(s);
for(i=1;i<=n;i++)m[i]=s[i-1]-'0';
mat a=mat(); a.size=n;
for(i=1;i<=n;i++)a.a[i][i]=a.a[i][i+1]=1; a.a[n][1]=1;
a=quickmulti(a,mul,MOD);
memset(ans,0,sizeof(ans));
for(i=1;i<=n;i++)
for(int j=1;j<=n;j++)
ans[i]=(ans[i]+m[j]*a.a[j][i])%MOD;
for(i=1;i<=n;i++)printf("%d",ans[i]); printf("\n");
}
return 0;
}