三元组
//三元组
#include<iostream>
#include<algorithm>
using namespace std;
const int N=210;
struct Node{
int row,col;
int v;
};
bool cmp(Node a,Node b){
if(a.row!=b.row) return a.row<b.row;
else return a.col<b.col;
}
class Tri{
private:
Node node[N];
int n,m;
int num;
public:
Tri();
Tri(int n,int m);
void set(int r,int c,int value);
void paixu();
void print1();
void print2();
int getValue(int r,int c);
int getn(){return n;}
int getm(){return m;}
};
Tri::Tri(){
int n=0,m=0,num=0;
}
Tri::Tri(int n,int m){
this->n=n;
this->m=m;
num=0;
}
void Tri::set(int r,int c,int value){
node[num].row=r;
node[num].col=c;
node[num].v=value;
num++;
}
void Tri::paixu(){
sort(node,node+num,cmp);
}
void Tri::print1(){
cout<<"num="<<num<<endl;
cout<<"行"<<" 列"<<" 值"<<endl;
for(int i=0;i<num;i++){
cout<<node[i].row<<" "<<node[i].col<<" "<<node[i].v<<endl;
}
}
void Tri::print2(){
for(int i=1,ct=0;i<=n;i++){
for(int j=1;j<=m;j++){
if(ct<num&&node[ct].row==i&&node[ct].col==j){
cout<<node[ct].v<<" ";
ct++;
}
else{
cout<<0<<" ";
}
}
cout<<endl;
}
}
int Tri::getValue(int r,int c){
for(int i=0;i<num;i++){
if(node[i].row==r&&node[i].col==c) return node[i].v;
}
return 0;
}
int main(){
int n,m,num;
cout<<"请输入第一个三元组的行、列和非零个数"<<endl;
cin>>n>>m>>num;
Tri T1(n,m);
cout<<"请输入"<<num<<"个非零元素的信息(行号列号和值)"<<endl;
int r,c,value;
for(int i=0;i<num;i++){
cin>>r>>c>>value;
T1.set(r,c,value);
}
T1.paixu();
cout<<"打印三元组"<<endl;
T1.print1();
T1.print2();
cout<<"现在开始矩阵相加。。。"<<endl<<"请输入第二个三元组非零个数(行列与T1相同)"<<endl;
cin>>num;
Tri T2(n,m);
cout<<"请输入"<<num<<"个非零元素的信息(行号列号和值)"<<endl;
for(int i=0;i<num;i++){
cin>>r>>c>>value;
T2.set(r,c,value);
}
T2.paixu();
cout<<"打印三元组"<<endl;
T2.print1();
T2.print2();
cout<<"相加之后"<<endl;
int v1,v2;
for(int i=1;i<=n;i++){
for(int j=1;j<=m;j++){
v1=T1.getValue(i,j);
v2=T2.getValue(i,j);
cout<<v1+v2<<" ";
}
cout<<endl;
}
cout<<"现在进行矩阵相乘!!!"<<endl<<"请输入第三个三元组的行、列和非零个数(保证该三元组的行与T1的列相等)"<<endl;
cin>>n>>m>>num;
Tri T3(n,m);
cout<<"请输入"<<num<<"个非零元素的信息(行号列号和值)"<<endl;
for(int i=0;i<num;i++){
cin>>r>>c>>value;
T3.set(r,c,value);
}
T3.paixu();
cout<<"打印三元组"<<endl;
T3.print1();
T3.print2();
cout<<"相乘后"<<endl;
for(int i=1;i<=T1.getn();i++){
for(int j=1;j<=T3.getm();j++){
int c=0;
for(int k=1;k<=T1.getm();k++){
c+=T1.getValue(i,k)*T3.getValue(k,j);
}
if(c>0) cout<<c<<" ";
else cout<<0<<" ";
}
cout<<endl;
}
}
这里简单汪两句:
1.涉及到一个sort()排序的自定义,代码如下
bool cmp(Node a,Node b){
if(a.row!=b.row) return a.row<b.row;
else return a.col<b.col;
}
void Tri::paixu(){
sort(node,node+num,cmp);
}
2.矩阵相乘的模板
for(int i=0;i<m;i++){
for(int j=0;j<l;j++){
int tmp=0;
for(int k=0;k<n;k++){
tmp+=a[i][k]*b[k][j];
}
c[i][j]=tmp;
}
}
满足条件:两个矩阵中其中一个的行等于另一个的列数。
大概就这些??