hdu 2276 Kiki & Little Kiki 2(矩阵优化递推)

hdu 2276 Kiki & Little Kiki 2

题目要求一个灯能开关的前提的条件是其左边那个灯是亮着的，先列出两个等在一起的所有状态，1表示亮着

0 0

0 1

1 0

1 1

共四种状态，观察一下，只有10和11两种状态的右边那个灯是可以关开的，因为左边那个灯是亮着的

（1+0）%2=1表示右边等被点着了

（1+1）%2=0右边等被关闭了

另外两种情况（0+0）%2=0，（0+1）%2=1，右边的灯开关状态都没变，符合要求，因此可以建立一个以灯个数n为大小的转移矩阵

第一组数据的矩阵：

1 0 0 0 0 0 1

1 1 0 0 0 0 0

0 1 1 0 0 0 0

0 0 1 1 0 0 0

0 0 0 1 1 0 0

0 0 0 0 1 1 0

0 0 0 0 0 1 1

其他类似，注意mod2就行了

#include<stdio.h>
#include<string.h>
#define MAXN 101
struct matrix
{
    int m[MAXN][MAXN];
};
int n;
matrix multi(const matrix &a,const matrix &b)
{
    matrix ans;
    memset(&ans,0,sizeof(ans));
    for(int i=0;i<n;i++)
        for(int k=0;k<n;k++)
            if(a.m[i][k])
                for(int j=0;j<n;j++)
                {
                    ans.m[i][j]+=a.m[i][k]*b.m[k][j];
                    ans.m[i][j]%=2;
                }
    return ans;
}
matrix pow(matrix a,int k)
{
    matrix ans;
    memset(&ans,0,sizeof(ans));
    for(int i=0;i<n;i++) ans.m[i][i]=1;
    while(k)
    {
        if(k&1) ans=multi(ans,a);
        a=multi(a,a);
        k>>=1;
    }
    return ans;
}
char s[MAXN];
int main()
{
    matrix a;
    int m;
    while(scanf("%d",&m)!=EOF)
    {
        scanf("%s",s);
        n=(int)strlen(s);
        memset(&a,0,sizeof(a));
        a.m[0][0]=a.m[0][n-1]=1;
        for(int i=1;i<n;i++)
            a.m[i][i]=a.m[i][i-1]=1;
        a=pow(a,m);
        int ans;
        for(int i=0;i<n;i++)
        {
            ans=0;
            for(int j=0;j<n;j++)
            {
                ans+=a.m[i][j]*(s[j]-'0');
                ans%=2;
            }
            printf("%d",ans);
        }
        printf("\n");
    }
    return 0;
}