基础练习 矩阵乘法
时间限制:1.0s 内存限制:512.0MB
问题描述
给定一个N阶矩阵A,输出A的M次幂(M是非负整数)
例如:
A =
1 2
3 4
A的2次幂
7 10
15 22
输入格式
第一行是一个正整数N、M(1<=N<=30, 0<=M<=5),表示矩阵A的阶数和要求的幂数
接下来N行,每行N个绝对值不超过10的非负整数,描述矩阵A的值
输出格式
输出共N行,每行N个整数,表示A的M次幂所对应的矩阵。相邻的数之间用一个空格隔开
样例输入
2 2
1 2
3 4
样例输出
7 10
15 22
分析:模板题,矩阵快速幂解决。
代码:
#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
struct Matrix {
int m;
int n;
vector<vector<int> > v;
Matrix() {
}
Matrix(int _m, int _n) {
m = _m;
n = _n;
v.resize(m);
for (int i = 0; i < m; i++) {
v[i].resize(n);
}
}
Matrix(int _m, int _n, int flag) {
m = _m;
n = _n;
v.resize(m);
for (int i = 0; i < m; i++) {
v[i].resize(n);
}
if (flag == 1) {
for (int i = 0; i < m; i++) {
v[i][i] = 1;
}
}
}
};
Matrix mul(Matrix a, Matrix b) {
Matrix res = Matrix(a.m, b.n);
for (int i = 0; i < a.m; i++) {
for (int j = 0; j < b.n; j++) {
for (int k = 0; k < a.n; k++) {
res.v[i][j] += a.v[i][k] * b.v[k][j];
}
}
}
return res;
}
int main() {
int N, M;
cin >> N >> M;
Matrix ma = Matrix(N, N);
for (int i = 0; i < N; i++) {
for (int j = 0; j < N; j++) {
cin >> ma.v[i][j];
}
}
Matrix res = Matrix(N, N, 1);
while (M > 0) {
if (M & 1) {
res = mul(res, ma);
}
ma = mul(ma, ma);
M >>= 1;
}
for (int i = 0; i < res.m; i++) {
for (int j = 0; j < res.n; j++) {
printf("%d%c", res.v[i][j], j == res.n - 1 ? '\n' : ' ');
}
}
return 0;
}