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

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

Time Limit: 1 Sec   Memory Limit: 128 MB
Submit: 1178   Solved: 750
[ 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, 

#include<bits/stdc++.h>
using namespace std;
class Matrix
{
private:
    int hang,lie;
    bool flag;
    int **ppp;
public:
    friend istream& operator>>(istream& in,Matrix & abc)
    {
        in>>abc.hang>>abc.lie;
        abc.ppp = new int*[abc.hang];
        for(int i=0;i<abc.hang;i++)
        abc.ppp[i] = new int[abc.lie];
        for(int i=0;i<abc.hang;i++)
        {
            for(int j=0;j<abc.lie;j++)
            {
                in>>abc.ppp[i][j];
            }
        }
        return in;
    }
    friend ostream& operator<<(ostream& ou,Matrix & abc)
    {
        if(abc.flag == false){
                ou<<"Error"<<endl;
                return ou;
        }
        for(int i = 0;i<abc.hang;i++)
        {
            for(int j=0;j<abc.lie;j++)
            {
                if(j==0)
                ou<<abc.ppp[i][j];
                else ou<<" "<<abc.ppp[i][j];
            }
            ou<<endl;
        }
       // ou<<endl;
        return ou;
    }
    Matrix operator+(Matrix &another)
    {
        Matrix abc;
        abc.flag = true;
        if(another.hang != hang||another.lie != lie)
        {
            abc.flag=false;
            return abc;
        }
        abc.hang = hang;
        abc.lie = lie;
        abc.ppp = new int*[hang];
        for(int i=0;i<abc.hang;i++)
        abc.ppp[i] = new int[lie];
        for(int i=0;i<hang;i++)
        {
            for(int j=0;j<lie;j++)
            {
                abc.ppp[i][j] = ppp[i][j]+another.ppp[i][j];
            }
        }
        return abc;
    }
    Matrix operator*(Matrix &another)
    {
         Matrix abc;
        abc.flag = true;
        if(another.hang != lie)
        {
            abc.flag=false;
            return abc;
        }
        abc.hang = hang;
        abc.lie = another.lie;
        abc.ppp = new int*[hang];
        for(int i=0;i<abc.hang;i++)
        abc.ppp[i] = new int[another.lie];
        for(int i=0;i<hang;i++)
        {
            for(int j=0;j<another.lie;j++)
            {
                abc.ppp[i][j] =0;
                for(int k=0;k<lie;k++)
                    abc.ppp[i][j]+=another.ppp[k][j]*ppp[i][k];
            }
        }
        return abc;
    }
};
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;
}
/**************************************************************
    Problem: 1816
    User: 201701060928
    Language: C++
    Result: Accepted
    Time:8 ms
    Memory:1268 kb
****************************************************************/