poj 3233 Matrix Power Series

http://poj.org/problem?id=3233
Description

Given a n × n matrix A and a positive integer k, find the sum S = A + A2 + A3 + … + Ak.

Input

The input contains exactly one test case. The first line of input contains three positive integers n (n ≤ 30), k (k ≤ 109) and m (m < 104). Then follow n lines each containing n nonnegative integers below 32,768, giving A’s elements in row-major order.

Output

Output the elements of S modulo m in the same way as A is given.

Sample Input

2 2 4
0 1
1 1
Sample Output

1 2
2 3
题目大意：给你一个矩阵A，算出 S = A + A^2 + A^3 + … + A^k.并输出矩阵S；

解题思路：由于这题的k比较的而时间只有3秒，除了算快速幂还要求累加，这样很容易超时，所以我们要想办法优化。这里我们可以用二分的方法来优化。
如：S(10) = A + A^2 + A^3 + A^4 + A^5+ A^6 + A^7 + A^8 + A^9+ A^10；
= S(5) + A^5 *S(5);
S(5) = A + A^2 + A^3 + A^4 + A^5;
= S(2) + A^3 + A^3*S(2);
S(2) = S(1) + A * S(1);
S(1) = A;
看到这，我们不难发现规律，那就是当k为奇数时s（k） = s（k/2）+A^(k/2+1)+A^(k/2+1)* s（k/2）;
当k为偶数时 s（k） = s（k/2）+A^(k/2)* s（k/2）;
利用这个规律就可以吧线性的时间复杂度简化为log（n）了。
具体代码实现：

#include <cstdio>
using namespace std;
struct Matrix
{
    int m[35][35];
}a;
int m;
Matrix Mult(Matrix a,Matrix b,int n)
{
    Matrix temp;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++)
            temp.m[i][j] = 0;
    }
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++)
        for(int k=1;k<=n;k++)
         temp.m[i][j] = (temp.m[i][j]+(a.m[i][k]*b.m[k][j])%m)%m;
    }
    return temp;
}
Matrix Add(Matrix a,Matrix b,int n)
{
    Matrix temp;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++)
            temp.m[i][j] = 0;
    }
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++)
            temp.m[i][j] = (a.m[i][j]+b.m[i][j])%m;
    }
    return temp;
}
Matrix QuickPower(Matrix a,int n,int k)
{
    Matrix ans,res;
    for(int i=1;i<=n;i++)
       for(int j=1;j<=n;j++)
       if(i==j)ans.m[i][j] = 1;
       else ans.m[i][j] = 0;
    res = a;
    while(k){
        if(k&1)
            ans = Mult(ans,res,n);
        res = Mult(res,res,n);
        k>>=1;
    }
    return ans;
}
Matrix work(int n,int k)
///核心代码，因为这里我们只要二分分到1时可以直接得到答案，所以直到k=1时返回到上一层，
///然后判断这一层的k是计数还是偶数，按照各自的规则算出他们的值然后再返回到上一层，直到结束，这样就可以算出s(k)了。
{
    if(k == 1)
        return a;
    Matrix res = work(n,k/2);
    Matrix temp;
    if(k&1){
        temp = Add(res,QuickPower(a,n,k/2+1),n);
        res = Add(temp,Mult(QuickPower(a,n,k/2+1),res,n),n);
    }
    else{
        res = Add(res,Mult(QuickPower(a,n,k/2),res,n),n);
    }
    return res;
}
int main()
{
    int n,k;
     scanf("%d %d %d",&n,&k,&m);
     for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++)
            scanf("%d",&a.m[i][j]);
     }
    Matrix S = work(n,k);
    for(int i=1;i<=n;i++)
    {
        printf("%d",S.m[i][1]);
        for(int j=2;j<=n;j++)
            printf(" %d",S.m[i][j]);
        printf("\n");
    }
    return 0;
}