北邮OJ 矩阵幂

题目描述

给定一个n*n的矩阵，求该矩阵的k次幂，即P^k

输入描述

第一行：两个整数n（2<=n<=10）、k（1<=k<=5），两个数字之间用一个空格隔开，含义如上所示。
接下来有n行，每行n个正整数，其中，第i行第j个整数表示矩阵中第i行第j列的矩阵元素Pij且（0<=Pij<=10）。另外，数据保证最后结果不会超过10^8。

输出描述

对于每组测试数据，输出其结果。格式为：
n行n列个整数，每行数之间用空格隔开，注意，每行最后一个数后面不应该有多余的空格

示例

输入

3
2 2
9 8
9 3
3 3
4 8 4
9 3 0
3 5 7
5 2
4 0 3 0 1
0 0 5 8 5
8 9 8 5 3
9 6 1 7 8
7 2 5 7 3

输出

153 96
108 81
1216 1248 708
1089 927 504
1161 1151 739
47 29 41 22 16
147 103 73 116 94
162 108 153 168 126
163 67 112 158 122
152 93 93 111 97

解题思路: 本题主要实现两个矩阵相乘即可，利用矩阵相乘法则，借助三重循环即可实现。

AC代码

#include<bits/stdc++.h>
using namespace std;
vector<vector<int> > getMatrix(vector<vector<int> >p,int n,int k)
{
    vector<vector<int> >m(n,vector<int>(n,0));
    vector<vector<int> >ans(n,vector<int>(n,0));
    for(int i=0;i<n;i++)
        m[i][i]=1;
    while(k--)
    {
        //m*p;
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<n;j++)
            {
                for(int k=0;k<n;k++)
                {
                    ans[k][i]+=m[k][j]*p[j][i];
                }
            }
        }
        m=ans;
        for(int i=0;i<n;i++)
            for(int j=0;j<n;j++)
                ans[i][j]=0;
    }
    return m;
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int n,k;
        scanf("%d%d",&n,&k);
        vector<vector<int> >p(n,vector<int>(n,0));

        for(int i=0;i<n;i++)
        {
            for(int j=0;j<n;j++)
                scanf("%d",&p[i][j]);
        }
        vector<vector<int> >ans=getMatrix(p,n,k);
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<n;j++){
                printf("%d ",ans[i][j]);
            }
            printf("\n");
        }
    }
    return 0;
}

记录

在本题中二维数组是利用C++中的vector实现的，vector相对于传统数组更灵活一些。

声明一个二维数组并赋初值vector<vector<int> >p(n,vector<int>(n,0));

还要注意的是resize()和reserve()的区别

1. resize()
   既分配空间，也创建了对象
2. reverse()
   仅表明了vector预留空间，但未创建对象，须通过insert和push_back操作创建对象