Problem A: 强悍的矩阵运算来了

Problem A: 强悍的矩阵运算来了

Time Limit: 1 Sec   Memory Limit: 128 MB
Submit: 677   Solved: 437
[ Submit][ Status][ Web Board]

Description

定义一个Matrix类，用于存储一个矩阵。重载其+、*运算符，分别用于计算两个矩阵的和、乘积；重载其<<和>>运算符，用于输出和输入一个矩阵。要求当两个矩阵不能进行加法或乘法运算时，应该输出Error。

Input

输入第1行N>0，表示有N组测试用例，共2N个矩阵。

每组测试用例包括2个矩阵。每个矩阵首先输入行数、列数，之后是该矩阵的所有元素。

Output

每个测试用例产生一组输出。具体格式见样例。注意：当不能进行加法或乘法运算时，应输出Error。

Sample Input

32 21 11 12 22 22 21 111 22 21 112 22 22 2

Sample Output

Case 1:3 33 34 44 4Case 2:Error2 2Case 3:ErrorError

HINT

Append Code

append.cc,

int main() {      int cases, i;      cin>>cases;      for (i = 0; i < cases; i++)      {          Matrix A, B, C, D;          cin>>A>>B;          C = A + B;          D = A * B;          cout<< "Case " <<i + 1<< ":" <<endl;          cout<<C<<endl;          cout<<D;      }      return 0; }

#include <iostream>
using namespace std;
class Matrix
{
public:
    int s[110][110];
    int l,v,flag;
public:
    Matrix()
    {
        flag=0;
    }
    Matrix operator + (const Matrix &a)
    {
        Matrix b;
        b.l=a.l;
        b.v=a.v;
        if(l!=a.l||v!=a.v)
        {
            b.flag=1;
            return b;
        }
        for(int i = 0; i<l; i++)
        {
            for(int j=0; j<v; j++)
            {
                b.s[i][j]=s[i][j]+a.s[i][j];
            }
        }
        return b;
    }
    Matrix operator * (Matrix &a)
    {
        Matrix b;
        b.l=l;
        b.v=a.v;
        for(int i=0; i<b.l; i++)
        {
            for(int j = 0; j<b.v; j++)
            {
                b.s[i][j]=0;
            }
 
        }
        if(v!=a.l)
        {
            b.flag = 1;
            return b;
        }
        for(int i=0; i<l; i++)
        {
            for(int j=0; j<a.v; j++)
            {
                int sum = 0;
                for(int q=0; q<a.l; q++)
                {
                    sum += s[i][q] * a.s[q][j];
                }
                b.s[i][j]  = sum;
            }
        }
        return b;
    }
    friend istream &operator>>(istream &is,Matrix &a)
    {
        is>>a.l>>a.v;
        for(int i=0; i<a.l; i++)
            {
                for(int j=0; j<a.v; j++)
            {
                is>>a.s[i][j];
            }
            }
        return is;
    }
 
    friend ostream &operator<<(ostream &os,Matrix &a)
    {
        if(a.flag==1)
        {
            os<<"Error"<<endl;
            return os;
        }
        for(int i=0; i<a.l; i++)
            for(int j=0; j<a.v; j++)
            {
                if(j!=a.v-1)
                    os<<a.s[i][j]<<" ";
                else
                    os<<a.s[i][j]<<endl;
            }
        return os;
    }
 
};
int main()
{
    int cases, i;
    cin>>cases;
    for (i = 0; i < cases; i++)
    {
        Matrix A, B, C, D;
        cin>>A>>B;
        C = A + B;
        D = A * B;
        cout<<"Case "<<i + 1<<":"<<endl;
        cout<<C<<endl;
        cout<<D;
    }
    return 0;
}