POJ 3233 Matrix Power Series

 转至:http://blog.csdn.net/wohenlaoshia/archive/2008/08/07/2784489.aspx

http://acm.pku.edu.cn/JudgeOnline/problem?id=3233

Matrix Power Series

Time Limit: 3000MS		Memory Limit: 131072K
Total Submissions: 1904		Accepted: 597

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

Source

POJ Monthly--2007.06.03, Huang, Jinsong

 

给定n*n矩阵，计算A + A^2 +....+A^n的值。

                              A        I                                      A^(k+1)      I+A+.....+A^k

构造矩阵 B   =                         ,易得B^(k+1) =                                                       ,所以问题转化为求B^(k+1),计算可知，在B(A,I,0,I)

                              0         I                                      0                  I

中，（0，I）不影响（A，I）的值，所以矩阵B可以简化为B（A，I），假设C（A,B)是C(A,B,0,I)的简化，则C的乘法可定义为C*C=(A^2,A*B+B),固由此可推出B（A，I）的乘法定义。

 

1. #include <iostream>
2. #include <string>
3. using namespace std;
4. int n, m;
6. struct multiply {
7.     int a[31][62];
9.     multiply operator*(const multiply & l) {
10.         multiply temp;
11.         int i, j, k;
12.         memset(temp.a, 0, sizeof (temp.a));
13.         for (i = 0; i < n; i++)
14.             for (j = 0; j < n; j++)
15.                 for (k = 0; k < n; k++)
16.                     temp.a[i][j] = (temp.a[i][j] + a[i][k] * l.a[k][j]) % m;
17.         for (i = 0; i < n; i++)
18.             for (j = n; j < 2 * n; j++) {
19.                 temp.a[i][j] = (temp.a[i][j] + l.a[i][j]) % m;
20.                 for (k = 0; k < n; k++)
21.                     temp.a[i][j] = (temp.a[i][j] + l.a[i][k] * a[k][j]) % m;
22.             }
23.         return temp;
24.     }
25. } K;
27. multiply pow(multiply & t, int k) {
28.     multiply temp;
29.     if (k == 1) return t;
30.     temp = t * t;
31.     if (k & 1) return pow(temp, k / 2) * t;
32.     else return pow(temp, k / 2);
33. }
35. int main() {
36.     multiply ss;
37.     int k, i, j;
38.     scanf("%d%d%d", &n, &k, &m);
39.     for (i = 0; i < n; i++) {
40.         for (j = 0; j < n; j++) {
41.             scanf("%d", &K.a[i][j]);
42.         }
43.     }
44.     for (i = 0, j = n; i < n; i++, j++) {
45.         K.a[i][j] = 1;
46.     }
47.     ss = pow(K, k + 1);
48.     for (i = 0, j = n; i < n; i++, j++) {
49.         ss.a[i][j]--;
50.     }
51.     for (i = 0; i < n; i++) {
52.         for (j = n; j < 2 * n - 1; j++)
53.             printf("%d ", ss.a[i][j] % m);
54.         printf("%d/n", ss.a[i][j] % m);
55.     }
56.     return 0;
57. }

Time: 141MS

Memory: 1104K

 

===============================================================================================

学习到了原来二分是好东东的说-_____________-

这里附上自己的代码,感受到了和大牛的差距也...

T.T偶500+MS啊,差距也....

1. #include <iostream>
2. using namespace std;
3. #define MAX_SIZE 30
4. int moudle,k,size;
5. struct Mat
6. {
7.     long long g[MAX_SIZE][MAX_SIZE];
8.     Mat operator+(const Mat&a)
9.         {
10.             Mat temp;
11.             int i,j;
12.             for(i=0;i<size;i++)
13.                 for(j=0;j<size;j++)
14.                     temp.g[i][j]=(g[i][j]+a.g[i][j])%moudle;
15.             return temp;
16.         }
17.     Mat operator*(const Mat&a)
18.         {
19.             Mat temp;
20.             int i,j,k;
21.             for(i=0;i<size;i++)
22.                 for(j=0;j<size;j++)
23.                 {
24.                     temp.g[i][j]=0;
25.                     for(k=0;k<size;k++)
26.                         temp.g[i][j]+=g[i][k]*a.g[k][j];
27.                     if(temp.g[i][j]>=moudle)
28.                         temp.g[i][j]%=moudle;
29.                 }
30.             return temp;
31.         }
32. }A,S;
33. Mat pow(int _k,Mat g)
34. {
35.     Mat temp;
36.         int i,j;
37.         for(i=0;i<size;i++)for(j=0;j<size;j++)if(i!=j)temp.g[i][j]=0;else temp.g[i][j]=1;
38.     for(;_k;g=g*g,_k>>=1)
39.           if(_k&1)temp=temp*g;
40.    return temp;
41. }
42. Mat sum(int _k)
43. {
44.     if(_k==1) return A;
45.     Mat temp,tpow;
46.     temp=sum(_k/2);
47.     if(_k&1)
48.         temp=temp+temp*(tpow=pow(_k/2+1,A))+tpow;
49.     else
50.         temp=temp+pow(_k/2,A)*temp;
51.     return temp;
52. }
53. int main()
54. {
55.     int i,j;
56.     scanf("%d%d%d",&size,&k,&moudle);
57.     for(i=0;i<size;i++)
58.         for(j=0;j<size;j++)
59.             scanf("%d",&A.g[i][j]);
60.     S=sum(k);
61.     for(i=0;i<size;i++)
62.     {
63.         for(j=0;j<size-1;j++)
64.             printf("%d ",S.g[i][j]);
65.         printf("%d/n",S.g[i][size-1]);
66.     }
67.     return 0;
68. }