二维向量的子向量

Question

我有std::vector<std::vector<double>> A(300, std::vector<double>(500)).

我想创建一个子范围为 A 的新向量：sub-vector[5:10][25:100]。我该怎么做？

Answer 1

您可以使用迭代器。首先创建向量：

std::vector<std::vector<double>> sub_vector;
sub_vector.reserve(5);

然后用向量的范围构造函数填充它：

for (std::size_t i = 5; i < 10; ++i) {
    sub_vector.emplace_back(A[i].begin() + 25, A[i].begin() + 100);
}

备注：

您有责任确保索引在范围内。否则会导致未定义的行为。
这涉及收/开范围。如果你想要close-close ranges，你需要在结束索引上加一。

Answer 2

没有像这样的简单符号，您必须自己动手：

std::vector<std::vector<double>> A(300, std::vector<double>(500));
std::vector<std::vector<double>> subranges;

subranges.reserve(11 - 5);
std::transform(A.begin() + 5, A.begin() + 11,
               std::back_inserter(subranges),
               [](const auto& inner){ // [](const std::vector<double>& inner) {
                   return std::vector<double>(inner.begin() + 25,
                                              inner.begin() + 101);
               });

Answer 3

另一种解决方案

使用函数完成工作。

#include <iostream>
#include <vector>
#include <iterator>
#include <algorithm>

constexpr size_t MaxRows = 300;
constexpr size_t MaxColumns = 500;
using MyType = double;
using Columns = std::vector<MyType>;
using Matrix = std::vector<Columns>;

void copySubMatrix( const Matrix& source,Matrix& destination,const size_t& startRow,const size_t& endRow,const size_t& startColumn, const size_t& endColumn)
{
    // Clear destination matrix
    destination.clear();
    // Copy rows end columns
    std::for_each(source.begin() + startRow, source.begin() + endRow + 1, [&](const Columns & c) {
        Columns row{ c.begin() + startColumn, c.begin() + endColumn + 1};
        destination.push_back(row); });
}

int main() {

    // Define source matrix with given size and empty destination matrix
    Matrix A(MaxRows, Columns(MaxColumns));
    Matrix result{};

    // Fill source matrix with running values
    std::for_each(A.begin(), A.end(), [i = 0](Columns & c) mutable {for (MyType& m : c) m = i++; });


    // Copy the given range to the destination matrix
    copySubMatrix(A, result, 5, 10, 25, 100);

    // Display destination matrix
    std::for_each(result.begin(), result.end(), [](const Columns & c) {
        std::copy(c.begin(), c.end(), std::ostream_iterator<MyType>(std::cout, " ")); std::cout << "\n"; });

    return 0;
}

Answer 4

您可以通过遍历一个维度来做到这一点：

std::vector<vector<double>> main_vector(300, std::vector<double>(500));
std::vector<vector<double>> sub_vector;
std::vector<double>::const_iterator first, last;

unsigned int x_pos_start=5, x_pos_end=10;
unsigned int y_pos_start=25, y_pos_end=100;

sub_vector.resize(x_pos_end - x_pos_start + 1);
for(size_t i=x_pos_start; i<=x_pos_end; ++i)
{
    first = main_vector[i].begin() + y_pos_start;
    last = main_vector[i].begin() + y_pos_end;
    sub_vector[i].insert(sub_vector[i].begin(), first, last);
}

二维向量的子向量

Sub-vector of 2D vector

c++

slice

stdvector