遍历 C++ 矩阵中的列

Question

我想遍历 std::vector<std::vector<int>> 矩阵中的单个行和列并获取它们的总和。

我知道我可以在嵌套循环中执行此操作，但这是我的问题。我可以使用

int val_sum = 0;
std::for_each(matrix_[row].begin(),matrix_[row].end(),[&](int x) {  val_sum += x;});

对于列以及如何做到这一点？

Answer 1

您可以嵌套两个 for_each。您只需要注意外部 for_each 的每个元素都将是另一个 vector<int>，一行。 [Demo]

#include <algorithm>  // for_each
#include <iostream>  // cout
#include <vector>

int main()
{
    std::vector<std::vector<int>> v{ {1, 2}, {3, 4}, {5, 6} };
    std::for_each(std::cbegin(v), std::cend(v), [](auto& row) {
        std::for_each(std::cbegin(row), std::cend(row), [](auto n) {
            std::cout << n << " ";
        });
        std::cout << "\n";
    });
}

如果要对矩阵的所有元素求和，可以使用accumulate 而不是内部for_each 循环。 [Demo]

#include <algorithm>  // for_each
#include <iostream>  // cout
#include <numeric>  // accumulate
#include <vector>

int main()
{
    int result{};
    std::vector<std::vector<int>> v{ {1, 2}, {3, 4}, {5, 6} };
    std::for_each(std::cbegin(v), std::cend(v), [&result](auto& row) {
        result += std::accumulate(std::cbegin(row), std::cend(row), 0);
    });
    std::cout << "sum = " << result << "\n";
}

甚至有两个嵌套 accumulate。 [Demo]

#include <iostream>  // cout
#include <numeric>  // accumulate
#include <vector>

int main()
{
    std::vector<std::vector<int>> v{ {1, 2}, {3, 4}, {5, 6} };
    auto result = std::accumulate(std::cbegin(v), std::cend(v), 0, [](auto total, auto& row) {
        return total + std::accumulate(std::cbegin(row), std::cend(row), 0);
    });
    std::cout << "sum = " << result << "\n";
}

如果（毕竟，我从一开始就误解了你，并且）你想添加给定列的所有元素。 [Demo]

#include <iostream>  // cout
#include <numeric>  // accumulate
#include <vector>

int sum_column(const std::vector<std::vector<int>>& v, size_t column)
{
    return std::accumulate(std::cbegin(v), std::cend(v), 0, [&column](auto total, auto& row) {
        return total + row[column];
    });
}

int main()
{
    std::vector<std::vector<int>> v{ {1, 2}, {3, 4}, {5, 6} };
    std::cout << "sum of column 1 = " << sum_column(v, 1) << "\n";
}

Answer 2

您提议的类似方法是迭代矩阵行并累积给定列中的元素。

int val_sum = 0;
std::for_each(matrix.begin(),matrix.end(),[&](std::vector<int> &row) {  val_sum += row[column];});

但我还是更喜欢使用 c++11 范围循环版本

int val_sum = 0;
for ( const std::vector<int> &row : matrix )
   val_sum += row[column];

Answer 3

您可以使用嵌套 std::accumulate:

#include <iostream>
#include <vector>
#include <numeric>

int main() {
    std::vector<std::vector<int>> matrix = { {1, 2, 3}, {2, 3, 4}, {3, 4, 5} };
    const int sum = std::accumulate(matrix.cbegin(), matrix.cend(), 0, [](const auto acc, const auto row) {
        return acc + std::accumulate(row.cbegin(), row.cend(), 0);
    });
    std::cout << sum;
}

遍历 C++ 矩阵中的列

Iterate over columns in C++ matrix

c++

vector

std