1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70
| #include<iostream> #include<vector> #include<algorithm> #include<cmath> #include<unordered_map>
using namespace std;
void mul(vector<vector<int>>& a, const vector<vector<int>> b, int n) { for(int i = 0; i < n; i++) { vector<int> ta; for(int j = 0; j < n; j++) { int m = 0; for(int k = 0; k < n; k++) { m += a[i][k] * b[k][j]; } ta.push_back(m); } a[i] = ta; } }
void matPow(int n, vector<vector<int>>& res, const vector<vector<int>>& arr, int pow) { switch(pow) { case 1: break; case 2: mul(res, arr, n); break; case 3: mul(res, arr, n); mul(res, arr, n); break; case 4: mul(res, arr, n); mul(res, res, n); break; case 5: mul(res, arr, n); mul(res, res, n); mul(res, arr, n); break; } } int main() { int n, k; while(scanf("%d%d", &n, &k) != EOF){ vector<vector<int>> arr(n, vector<int>(n, 0)); vector<vector<int>> res(n, vector<int>(n, 0)); for(int i = 0; i < n; i++) { for(int j = 0; j < n; j++) { int t; scanf("%d", &t); res[i][j] = t; arr[i][j] = t; } } matPow(n, res, arr, k); for(int i = 0; i < n; i++) { for(int j = 0; j < n-1; j++) { printf("%d ", res[i][j]); } printf("%d\n", res[i][n-1]); } } return 0; }
|