浙江理工大学入队200题——12C

问题 C: 零基础学C/C++128——矩阵计算

题目描述

给定2个rc阶矩阵m1、m2以及若干运算符包括：+、-、、=，分别代表加(m1+m2)、减(m1-m2)、乘(m1*m2)、赋值(m1=m2)操作。计算矩阵的运算结果。其中1<=r,c<=10。

输入

输入数据有多组，第一行为测试数据的组数T,每组测试数据由2个矩阵（m1和m2）和若干个运算符组成。每个矩阵的第一行是矩阵的行、列数r，c。后面有r行，每行c个整数。
运算以end结束。我们假设所有的运算操作均可以进行

输出

赋值运算不输出任何结果，其他运行均输出每次运算的结果，且每次输出结果矩阵后再输出一个空行。

样例输入 Copy

1
3 3
1 0 0
0 1 0
0 0 1
3 3
1 2 3
1 2 3
1 2 3

=
+
end

样例输出 Copy

0 -2 -3
-1 -1 -3
-1 -2 -2

1 2 3
1 2 3
1 2 3

2 4 6
2 4 6
2 4 6

题解

本身是比较复杂的过程，分为四个小块围绕+-*/来写，没有简单方法
为了省时间我用c++来写了，如果用c的话可以对应着模拟

代码（AC）

点击查看代码

#include <iostream>
#include <string>
#include <cstring>
using namespace std;
int main()
{
	int r,c,r1,c1;
	int a[110][110],b[110][110],ans[110][110];
	int t;
	memset(a,0,sizeof(a));
	memset(b,0,sizeof(b));
	memset(ans,0,sizeof(ans));
	cin>>t;
	while (t--)
	{
		cin>>r>>c;
		for(int i=0;i<r;i++)
		{
			for(int j=0;j<c;j++)
			{
				cin>>a[i][j];
			}
		}
		cin>>r1>>c1;
		for(int i=0;i<r1;i++)
		{
			for(int j=0;j<c1;j++)
			{
				cin>>b[i][j];
			}
		}
		string ys;
		while(cin>>ys&&ys!="end")
		{
			if(ys=="-")
			{
				for(int i=0;i<r;i++)
				{
					for(int j=0;j<c;j++)
					{
						ans[i][j]=a[i][j]-b[i][j];
					}
				}
				for(int i=0;i<r;i++)
				{
					for(int j=0;j<c;j++)
					{
						cout<<ans[i][j];
						if(j!=c-1) cout<<" ";
					}
					cout<<endl;
				}
				cout<<endl;
			}
                        else if(ys=="+")
			{
				for(int i=0;i<r;i++)
				{
					for(int j=0;j<c;j++)
					{
						ans[i][j]=a[i][j]+b[i][j];
					}
				}
				for(int i=0;i<r;i++)
				{
					for(int j=0;j<c;j++)
					{
						cout<<ans[i][j];
						if(j!=c-1) cout<<" ";
					}
					cout<<endl;
				}
				cout<<endl;
			}
                        else if(ys=="=")
			{
				for(int i=0;i<r;i++)
				{
					for(int j=0;j<c;j++)
					{
						a[i][j]=b[i][j];
					}
				}
			}
                        else if(ys=="*")
			{
				memset(ans,0,sizeof(ans));
				for(int i=0;i<r;i++)
				{
					for(int j=0;j<c;j++)
					{
						for(int k=0;k<r;k++)
						{
							ans[i][j]+=a[i][k]*b[k][j];
						 } 
					}
				}
				for(int i=0;i<r;i++)
				{
					for(int j=0;j<c;j++)
					{
						cout<<ans[i][j];
						if(j!=c-1) cout<<" ";
					}
					cout<<endl;
				}
				cout<<endl;
			}
		}
	}
    return 0;
}