poj 3233 Matrix Power Series

题目链接： http://poj.org/problem?id=3233

这道题我个人认为是一道中等难度的矩阵快速幂吧！（中等对我来说就是depp难了，想了好久==）。

思路：我是运用了矩阵可以作为矩阵的元素的思想找到了一个表达式：

根据这个公式， 我构造了一个 num矩阵和ans矩阵， 这道题就转化成了对 num矩阵的快速幂问题。 下面是代码:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
using namespace std;
typedef long long ll;
struct matrix
{
   int  arr[70][70];
};
matrix chengfa(matrix a, matrix b, ll n, ll m);
matrix qiumi(matrix a, ll b,ll n, ll m);
int main()
{
    ll n,m,k;
    while(~scanf("%lld%lld%lld",&n,&k,&m))
    {
        int i,j; matrix num,ans;
        memset(num.arr,0,sizeof(num.arr));
        for(i=1; i<=n; i++)
          for(j=1; j<=n; j++)
             scanf("%d",&num.arr[i][j]);

        ans = num;

        for(i=n+1; i<=n<<1; i++)
           for(j=1; j<=n; j++)
              ans.arr[i][j] = (i-n == j);  //构造ans矩阵

        for(i=1; i<=n; i++)
           for(j=n+1; j<=n*2; j++)
                num.arr[i][j] = num.arr[i][j-n];
        for(i=n+1; i<=n<<1; i++)
            for(j=n+1; j<=n<<1; j++)
                num.arr[i][j] = (i==j); //构造num矩阵

        num = qiumi(num,k-1,n*2,m);    //对num矩阵进行求幂运算

        matrix anss;
        memset(anss.arr,0,sizeof(anss));
        for(i=1; i<=n*2; i++)
        {
            for(j=1; j<=n; j++)
                for(k=1; k<=n*2; k++)
                    anss.arr[i][j] += ((num.arr[i][k]%m)*(ans.arr[k][j]%m))%m;  //最后再相乘一下就好了。
        }
        for(i=1; i<=n; i++)
        {
            for(j=1; j<=n; j++)
                if(j == 1)
                    printf("%d",anss.arr[i][j]%m);
                else
                    printf(" %d",anss.arr[i][j]%m);
            printf("\n");
        }
    }
    return 0;
}
matrix chengfa(matrix a, matrix b, ll n, ll m)
{
    matrix c;
    memset(c.arr,0,sizeof(c.arr));
    for(int i=1; i<=n; i++)
       for(int j=1; j<=n; j++)
         for(int k=1; k<=n; k++)
            c.arr[i][j] += ((a.arr[i][k]%m) * (b.arr[k][j]%m))%m;
    return c;
}
matrix qiumi(matrix a, ll b,ll n, ll m)
{
    matrix ans;
    int i,j,k;
    for(i=1; i<=n; i++)
        for(j=1; j<=n; j++)
            ans.arr[i][j] = (i==j); //单位矩阵
    while(b > 0)
    {
        if(b&1)
            ans = chengfa(ans,a,n,m);
        b /= 2;
        a = chengfa(a,a,n,m);
    }
    return ans;
}

我最初想的是 这道题的表达式是一个等比数列， 然后可以直接用前n项和求得。 但是我还不用c/c++求矩阵的逆矩阵，只好暂时作罢。这个代码运行了2344ms 我想肯定还有更加简便的写法，不过这道题让我学到了矩阵里面套矩阵的算法思想。