首页 > 编程语言 > 详细

二维矩阵相乘 in C++

时间:2019-02-28 11:15:35      阅读:179      评论:0      收藏:0      [点我收藏+]

技术分享图片

#include <iostream>
#include <vector>
#include <string>
#include <sstream>
#include <fstream>
#include <algorithm>
#include <functional>
#include <numeric>

template <class DataType>
void ReadMatFromFile(std::string &filename, std::vector<std::vector<DataType> > &lines_feat) {
    std::ifstream vm_info(filename.c_str());
    std::string lines;
    DataType var;
    std::vector<DataType> row;

    lines_feat.clear();

    while(!vm_info.eof()) {
        getline(vm_info, lines);
        if(lines.empty())
            break;
        std::stringstream stringin(lines);
        row.clear();

        while(stringin >> var) {
            row.push_back(var);
        }
    lines_feat.push_back(row);
    }
}

template <class T1, class T2>
int MatMultiply(const std::vector<std::vector<T1> > &Mata, const std::vector<std::vector<T2> > &Matb, std::vector<std::vector<T1> > &MatOut) {
    if(Mata.at(0).size() != Matb.size()) {
        std::cout<<"not match!\n";
        return -1;
    }
    for(size_t i=0; i<Mata.size(); ++i) {
        for(size_t j=0; j<Matb.at(0).size(); ++j) {
            std::vector<T2> col;
            col.clear();
                for(size_t k=0; k<Matb.size(); ++k) {
                    col.push_back(Matb.at(k).at(j));
                }
            MatOut.at(i).at(j)=inner_product(Mata.at(i).begin(), Mata.at(i).end(), col.begin(), 0);
        }
    }
    return 0;
}

template <class T>
void Display2DVector(std::vector<std::vector<T> > &vv) {
    for(size_t i=0;i<vv.size();++i) {
        for(typename::std::vector<T>::const_iterator it=vv.at(i).begin();it!=vv.at(i).end();++it) {
            std::cout<<*it<<" ";
        }
        std::cout<<"\n";
    }
    std::cout<<"--------the total of the 2DVector is "<<vv.size()<<std::endl;
}

int main() {
    std::string matfa="mata.dat", matfb="matb.dat";

    std::vector<std::vector<int> > mata, matb;

    ReadMatFromFile(matfa, mata);

    Display2DVector(mata);

    ReadMatFromFile(matfb, matb);

    Display2DVector(matb);

    std::cout<<mata.size()<<"\t"<<matb.at(0).size()<<std::endl;

    std::vector<std::vector<int> > matout(mata.size(), std::vector<int>(matb.at(0).size()));

    MatMultiply(mata, matb, matout);

    Display2DVector(matout);

    return 0;

二维矩阵相乘 in C++

原文:https://www.cnblogs.com/donggongdechen/p/10448868.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!