c++ - 在 vector vector 的 vector 中找到最大位置

Question

我有一个 vector 的 vector 的 vector

std::vector<std::vector<std::vector<double>>> mountain_table

我想找到该 vector 的最高坐标 i、j、k。我知道我应该使用 max_element 但我不知道如何在 3d vector 中使用它。

我应该如何获得这些坐标？

Answer 1

我建议将您的数据线性化，以便能够使用标准算法。这个想法是提供几个函数来从 3D 坐标获取索引，反之亦然:

template<class T>
class Matrix3D // minimal
{
public:
    using value_type = T;
    using iterator   = std::vector<value_type>::iterator;

private:
    std::vector<value_type> _data;
    size_t _sizex, _sizey, _sizez;

    size_t index_from_coords(size_t x, size_t y, size_t z) const
    {
        return x*_sizex*_sizey + y*_sizey + z;
    }
    std::tuple<size_t, size_t, size_t> coords_from_index(size_t index) const
    {
        const size_t x = index / (_sizex * _sizey);
        index = index % x;
        const size_t y = index / _sizey;
        const size_t z = index % _sizey;
        return make_tuple(x, y, z);
    }

public:
    Matrix3D(size_t sizex, sizey, sizez) : _sizex(sizex), ... {}
    T& operator()(size_t x, size_t y, size_t z) // add const version
    {
        return _data[index_from_coords(x, y, z)];
    }
    std::tuple<size_t, size_t, size_t> coords(iterator it)
    {
        size_t index = std::distance(begin(_data), it);
        return coords_from_index(index);
    }
    iterator begin() { return begin(_data); }
    iterator end()   { return end(_data);   }
}

用法:

Matrix3D<double> m(3, 3, 3);
auto it = std::max_element(m.begin(), m.end()); // or min, or whatever from http://en.cppreference.com/w/cpp/header/algorithm
auto coords = m.coords(it);
std::cout << "x=" << coords.get<0>() << ... << "\n";

这是未经测试且不完整的代码，可让您快速开始更好的数据设计。我很乐意在下面的评论中回答关于这个想法的更多问题；)