【POJ 3233】Matrix Power Series

【题目链接】

          点击打开链接

【算法】

          要求 A^1 + A^2 + A^3 + ... + A^k

          考虑通过二分来计算这个式子 ：

          令f(k) = A^1 + A^2 + A ^ 3 + ... + A^k

          那么，当k为奇数时,f(k) = f(k-1) + A ^ k

                    当k为偶数时,f(k) = f(n/2) + A ^ (n/2) * f(n/2）

          因此，可以通过二分 + 矩阵乘法快速幂的方式，在O(n^3log(n)^2)的时间内解决此题

【代码】

         

#include <algorithm>
#include <bitset>
#include <cctype>
#include <cerrno>
#include <clocale>
#include <cmath>
#include <complex>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <ctime>
#include <deque>
#include <exception>
#include <fstream>
#include <functional>
#include <limits>
#include <list>
#include <map>
#include <iomanip>
#include <ios>
#include <iosfwd>
#include <iostream>
#include <istream>
#include <ostream>
#include <queue>
#include <set>
#include <sstream>
#include <stdexcept>
#include <streambuf>
#include <string>
#include <utility>
#include <vector>
#include <cwchar>
#include <cwctype>
#include <stack>
#include <limits.h>
using namespace std;
#define MAXN 35

int i,j,n,k,m;
struct Matrix
{
		int mat[MAXN][MAXN];
} a,ans;

inline Matrix add(Matrix a,Matrix b)
{
		int i,j;
		Matrix ans;
		memset(ans.mat,0,sizeof(ans.mat));
		for (i = 1; i <= n; i++)
		{
				for (j = 1; j <= n; j++)
				{
						ans.mat[i][j] = (a.mat[i][j] + b.mat[i][j]) % m;
				}
		}
		return ans;
}
inline Matrix mul(Matrix a,Matrix b)
{
		int i,j,k;
		Matrix ans;
		memset(ans.mat,0,sizeof(ans.mat));
		for (i = 1; i <= n; i++)
		{
				for (j = 1; j <= n; j++)
				{
						for (k = 1; k <= n; k++)
						{
								ans.mat[i][j] = (ans.mat[i][j] + a.mat[i][k] * b.mat[k][j]) % m;
						}
				}
		}
		return ans;
}
inline Matrix power(Matrix a,int m)
{
		Matrix ans,p = a;
		for (i = 1; i <= n; i++)
		{
				for (j = 1; j <= n; j++)
				{
						ans.mat[i][j] = (i == j);	
				}	
		}	
		while (m > 0)
		{
				if (m & 1) ans = mul(ans,p);
				p = mul(p,p);
				m >>= 1;
		}
		return ans;
}
Matrix solve(int n)
{
		Matrix tmp;
		if (n == 1) return a;
		if (n % 2 == 0)
		{
				tmp = solve(n/2);
				return add(tmp,mul(power(a,n/2),tmp));
		} else return add(solve(n-1),power(a,n));
 }

int main() 
{
		
		scanf("%d%d%d",&n,&k,&m);
		for (i = 1; i <= n; i++)
		{
				for (j = 1; j <= n; j++)
				{
						scanf("%d",&a.mat[i][j]);
				}
		}
		ans = solve(k);
		for (i = 1; i <= n; i++)
		{
				for (j = 1; j < n; j++)
				{
						printf("%d ",ans.mat[i][j]);
				}
				printf("%d\n",ans.mat[i][n]);
		}
		
		return 0;
	
}