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;
}