c++使用vector实现按列求和

最新推荐文章于 2023-12-06 08:52:23 发布

焦志坤

最新推荐文章于 2023-12-06 08:52:23 发布

阅读量1.6k

点赞数 1

分类专栏： C++学习之路

本文链接：https://blog.youkuaiyun.com/cute_boy_/article/details/119572026

版权

C++学习之路专栏收录该内容

17 篇文章

订阅专栏

c++使用vector实现按列求和

直接上代码

#include <vector>
#include <iostream>
using namespace std;


vector<vector<double>> colwise_sum(vector<vector<double>> m) {
    //矩阵按列求和
    vector<vector<double>> array;
    vector<double> tmp;
    for (int i = 0; i < m[0].size(); i++)
    {
        double sum = 0;
        for (int j = 0; j < m.size(); j++)
        {
            sum += m[j][i];
        }
        tmp.push_back(sum);
    }
    array.push_back(tmp);
    return array;
}


int main() {
    vector<vector<double>> nums1 = {
        {1, 2, 3},
        {4, 15, 6},
        {7, 8, 8}
    };
    vector<vector<double>> w = colwise_sum(nums1);
    for (int i = 0; i < w.size(); i++)
    {
        for (int j = 0; j < w[0].size(); j++)
        {
            cout << w[i][j] << " ";
        }
        cout << endl;
    }
}