如何使用枚举有效地填充2D std :: array

时间:2017-05-12 02:12:23

标签: c++ arrays enums c++14 stdarray

我正在尝试找到一种有效且正确的方法来填充具有std::array值的2D enum矩阵。我这样做:

#include <iostream>
#include <array>

template<class T, size_t ROW, size_t COL>
using Matrix = std::array<std::array<T, COL>, ROW>;

enum class State { FREE = 0, BUSY, BLOCKED, RESERVED };

int main() {
  const int mapX = 4;
  const int mapY = 9;

  // create a 5x10 2D array
  Matrix<State, mapY, mapX> MapMatrix;

  // fill array with State::RESERVED value
  for (int y = 0; y <= mapY; y++) MapMatrix[y].fill(State::RESERVED);

  std::cout << "MapMatrix contains:\n";

  for (int y = 0; y <= mapY; ++y) {
    for (int x = 0; x <= mapX; ++x) {
      std::cout << static_cast<int>(MapMatrix[x][y]) << " ";
    }
    std::cout << std::endl;
  }

  return 0;
}

for loop 我是否正在使用enum值填充矩阵的最佳方法?有没有办法在Matrix<State, mapY, mapX> MapMatrix(像构造函数)声明期间填充矩阵?

谢谢!

2 个答案:

答案 0 :(得分:3)

除非用零填充,否则无法在初始化时填充它,或者明确指定每个元素。如果您重新排列枚举以使RESERVED为零,那么您可以像这样初始化它:

Matrix<State, mapY, mapX> MapMatrix = {};

如果你不能这样做,那么是的,for循环可能是最好的选择。把事情简单化。但是你有一些问题。首先,您的评论说您正在创建一个5x10阵列,但事实并非如此。您正在创建一个4x9阵列。如果要创建5x10阵列,则需要传递5和10作为模板参数。我认为您可能会对N大小为N的数组的最后一个元素感到困惑。这个一个一个问题只与访问数组元素有关,而不是指定大小一个数组。

其次,您正在迭代数组的末尾,因为您的循环条件是y <= mapY,而不是y < mapY。但是如果你只是使用了一个range-for循环会更好。

for (auto& arr : MapMatrix)
    arr.fill(State::RESERVED);

答案 1 :(得分:2)

我认为基于循环的初始化是一个很好的解决方案。

但是,为了好玩,我建议您另一个解决方案std::index_sequence和基于模板包扩展。

一个工作示例(已更正索引)

#include <iostream>
#include <utility>
#include <array>

template <typename T, std::size_t ROW, std::size_t COL>
using Matrix = std::array<std::array<T, COL>, ROW>;

enum class State { FREE = 0, BUSY, BLOCKED, RESERVED };

template <typename T, std::size_t ... Rs, std::size_t ... Cl>
Matrix<T, sizeof...(Rs), sizeof...(Cl)> initMat
   (T                          const & tVal,
    std::index_sequence<Rs...> const &,
    std::index_sequence<Cl...> const &)
 {
   auto col = std::array<T, sizeof...(Cl)>{ { ((void)Cl, tVal)... } };

   return Matrix<T, sizeof...(Rs), sizeof...(Cl)>
    { { ((void)Rs, col)... } };
 }

int main()
 {
   constexpr std::size_t mapX =  5U;
   constexpr std::size_t mapY = 10U;

   // create a 5x10 2D array
   auto MapMatrix = initMat(State::RESERVED,
                            std::make_index_sequence<mapX>(),
                            std::make_index_sequence<mapY>());

   std::cout << "MapMatrix contains:\n";

   for ( auto y = 0U ; y < mapY ; ++y )
    {
      for ( auto x = 0U ; x < mapX ; ++x )
         std::cout << static_cast<int>(MapMatrix[x][y]) << " ";

      std::cout << std::endl;
    }

   return 0;
 }