POJ 3233 矩阵的幂（求和）

Matrix Power Series

Time Limit: 3000MS		Memory Limit: 131072K
Total Submissions: 18904		Accepted: 7991

Description

Given a n × n matrix A and a positive integer k, find the sum S = A + A² + A³ + … + A^k.

Input

The input contains exactly one test case. The first line of input contains three positive integers n (n ≤ 30), k (k ≤ 10⁹) and m (m < 10⁴). 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的n*n的放成，计算出累加和，然后求余即可。

遇到的问题和思路：

      首先看到这道题和单纯的矩阵的幂的3070感觉不同，如果要一个一个都计算了，那么要O（k * n * n * n），对于如此的数据肯定是超时。然后看了一下程序挑战的p204页。。。就是重新创建一个矩阵，是由A矩阵，0矩阵和两个单位矩阵构成的，然后每一次把和都放在左下角的矩阵中就可以了。具体的看书吧^0^。

#include<cstdio> #include<vector> #include<algorithm> using namespace std; typedef vector<int> vec; typedef vector<vec> mat; typedef long long ll; ll n, k, m; mat mul(mat &a, mat & b, ll mod){ mat c(a.size(), vec(a.size())); for (int i = 0; i < a.size(); i++){ for (int j = 0; j < b.size(); j++){ for (int k = 0; k < b[0].size(); k++){ c[i][j] = (c[i][j] + a[i][k] * b[k][j]) % mod; } } } return c; } mat pow_mod(mat a, ll n, ll mod){ mat c(a.size(), vec(a.size())); for (int i = 0; i < a.size(); i++){ c[i][i] = 1; } while (n > 0){ if (n & 1)c = mul(a, c, mod); a = mul(a, a, mod); n >>= 1; } return c; } void solve(){ mat a(n, vec(n)); for (int i = 0; i < n; i++){ for (int j = 0; j < n; j++){ scanf("%d", &a[i][j]); } } mat b(n * 2, vec(n * 2)); for (int i = 0; i < n; i++){ for (int j = 0; j < n; j++){ b[i][j] = a[i][j]; } b[i + n][i] = b[i + n][i + n] = 1; } b = pow_mod(b, k + 1, m); /* for (int i = 0; i < n; i++){ for (int j = 0; j < n; j++){ printf("%d ", b[n + i][j]); } printf("\n"); }*/ for (int i = 0; i < n; i++){ for (int j = 0; j < n; j++){ int A = b[n + i][j] % m; if (i == j) A = (A + m - 1) % m;//这里不能写成A = (b[n + 1][j] - 1) % m或者A = (b[n + 1][j] % m) - 1;因为当b的右下角是0的时候，-1就是负数。然而题目中A的每一项都>0，故一定为正 printf ("%d%c", A, j < n - 1? ' ':'\n'); } } } int main(){ while(scanf("%I64d%I64d%I64d", &n, &k, &m) != EOF){ solve(); } return 0; } </vec></int></algorithm></vector></cstdio>