矩阵快速幂

矩阵快速幂和普通快速幂写法相似。

struct Matrix
{
    int maze[maxn][maxn];

    void Init()
    {
        for(int i=0;i<n;i++)
            for(int j=0;j<n;j++)
                maze[i][j]=(i==j);
    }

    friend Matrix operator*(Matrix a,Matrix b)
    {
        Matrix c;
        memset(c.maze,0,sizeof c.maze);
        for(int i=0;i<n;i++)
            for(int j=0;j<n;j++)
                for(int k=0;k<n;k++)
                    c.maze[i][j]=(c.maze[i][j]+a.maze[i][k]*b.maze[k][j])%MOD;
        return c;
    }
    friend Matrix operator^(Matrix a,int k)
    {
        Matrix c;
        c.Init();
        while(k)
        {
            if(k&1)
                c=c*a;
            a=a*a;
            k>>=1;
        }
        return c;
    }
};

水题,练习下hdu 1575


你可能感兴趣的:(矩阵快速幂)