稀疏矩阵相乘mmult

系列文章目录

前言

《数据结构基础》c语言版 第2版，Ellis Horowitz著，朱仲涛译
1.3节，page64，程序2.10

一、题目描述

稀疏矩阵相乘

二、c++代码

代码如下：

#include<iostream>
using namespace std;

struct term {
    int row;           //行，下标为0时指矩阵的行数
    int col;           //列，下标为0时指矩阵的列数
    int value;         //数值，下标为0时指矩阵的非零元素数
};

term
        A[100] = { {3,4,6},{0,0,1},{0,2,2},{1,1,3},{1,3,4},{2,0,5},{2,2,6} },
        B[100] = { {4,3,4},{0,0,1},{1,1,1},{2,2,1},{3,0,1} };

//插入元素到积矩阵
void storesum ( term d[ ], int *t, int row, int col, int *sum) {
    if( *sum ) {
        d[++*t].row = row;
        d[*t].col = col;
        d[*t].value = *sum;
        *sum = 0;
    }
}

int compare(int a,int b)
{
    if(a > b)
        return 1;
    else if (a < b)
        return -1;
    else
        return 0;
}

//矩阵转置， x转置后得到new_b
void fast_transpose ( term x[ ], term new_b[ ] ) { //
    int start_pos[10];                 //转置后各行起始位置
    int row_terms[10];             //转置后的每行中非零元素的个数

    new_b[0].row = x[0].col;
    new_b[0].col = x[0].row;
    new_b[0].value = x[0].value;

    if( x[0].value > 0 ) {
        for( int i = 0; i < x[0].col; i++ )
            row_terms[i] = 0;
        for( int i = 1; i <= x[0].value; i++ )
            row_terms[x[i].col]++;
        start_pos[0] = 1;
        for( int i = 1; i < x[0].col; i++ )
            start_pos[i] = start_pos[i-1] + row_terms[i-1];
        for( int i = 1; i <= x[0].value; i++ ) {
            int j = start_pos[x[i].col];
            new_b[j].row = x[i].col;
            new_b[j].col = x[i].row;
            new_b[j].value = x[i].value;
            start_pos[x[i].col]++;
        }
    }
}

//矩阵a乘于矩阵b得到矩阵d;
void mmult ( term a[ ], term b[ ], term d[ ] ) {
    term new_b[100];
    fast_transpose ( b, new_b );   //转置第二个矩阵
    int sum = 0;
    int t = 0;                     //d总的元素个数
    //下面几行只是为了第二个for循环多循环一次,以致得到最后的那个结果
    new_b[b[0].value+1].row = b[0].col;
    new_b[b[0].value+1].col = 0;
    new_b[b[0].value+1].value = 0;
    a[a[0].value+1].row = a[0].row;

    for( int i = 1; i <= a[0].value; ) {
        int row = a[i].row;
        int column = new_b[1].row;
        int row_begin = i;      //记录当前行最前的非零元素的位置
        for(int j = 1; j <= b[0].value+1; ) {
            if( a[i].row != row ) {   //检查a矩阵当前行是否还有元素，如果没有 证明当前行非零元素已经遍历完，可以将结果存到d矩阵了
                storesum ( d, &t, row, column, &sum );
                i = row_begin;     //行中第一个非0数
                for( ; new_b[j].row == column; j++ ); //a中的当前行非零元素已经没有的，而b中还有
                column = new_b[j].row; //指向下一个(下一列那个)元素
            } else if( new_b[j].row != column ) {  //检查b矩阵当前列是否还有元素，如果没有 证明当前列非零元素已经遍历完，可以将结果存到d矩阵了
                storesum ( d, &t, row, column, &sum );
                i = row_begin;//a矩阵回到当前行的最前面一个非零元素，以致与new_b的下一行相乘
                column = new_b[j].row;//指向一个(下一列那个)元素
            } else {
                switch (compare(a[i].col,new_b[j].col)) {  //检查a的列是不是等于b的行
                    case -1:
                        i++;
                        break;
                    case 1:
                        j++;
                        break;
                    case 0:
                        sum+=(a[i].value*new_b[j].value);
                        i++;
                        j++;
                        break;
                }
            }
        }
        for( ; a[i].row == row; i++ );//b中的当前行非零元素已经没有的，而a中还有；
    }
    d[0].row = a[0].row;  //行数
    d[0].col = b[0].col;  //列数
    d[0].value = t;       //非零元素数目
}
int main() {
    term d[100];          //积
    mmult( A, B, d );
    for(int i=0; i<=d[0].value; i++)
        cout<<d[i].row<<' '<<d[i].col<<' '<<d[i].value<<endl;
    cout<<endl;
    return 0;
}

总结

添加链接描述

添加链接描述