UE4【C++】【实现2维和3维数组】

矫情吗；* 2023-03-01 14:57 6阅读 0赞

template <typename T>
    class Array2D
    {
    public:
    	Array2D(unsigned int _length, unsigned int _width, const T& defaultValue)
    		: length(_length)
    		, width(_width)
    	{
    		for (unsigned int x = 0; x < length; ++x)
    		{
    			matrix.Add(TArray<T>());
    
    			for (unsigned int y = 0; y < width; ++y)
    			{
    				matrix[x].Add(defaultValue);
    			}
    		}
    	}
    
    	inline TArray<T>& operator[](unsigned int x)
    	{
    		return matrix[x];
    	}
    
    	inline const TArray<T>& operator[](unsigned int x) const
    	{
    		return matrix[x];
    	}
    
    	inline const unsigned int getLength() const { return length; }
    	inline const unsigned int getWidth() const { return width; }
    
    private:
    	TArray<TArray<T>> matrix;
    	const unsigned int length;
    	const unsigned int width;
    };

template <typename T>
    class Array3D
    {
    public:
    	Array3D(unsigned int _length, unsigned int _width, unsigned int _height, const T& defaultValue)
    		: length(_length)
    		, width(_width)
    		, height(_height)
    	{
    		for (unsigned int x = 0; x < length; ++x)
    		{
    			matrix.Add(TArray<TArray<T>>());
    
    			for (unsigned int y = 0; y < width; ++y)
    			{
    				matrix[x].Add(TArray<T>());
    
    				for (unsigned int z = 0; z < height; ++z)
    				{
    					matrix[x][y].Add(defaultValue);
    				}
    			}
    		}
    	}
    
    	inline TArray<TArray<T>>& operator[](unsigned int x)
    	{
    		return matrix[x];
    	}
    
    	inline const TArray<TArray<T>>& operator[](unsigned int x) const
    	{
    		return matrix[x];
    	}
    
    	inline const unsigned int getLength() const { return length; }
    	inline const unsigned int getWidth() const { return width; }
    	inline const unsigned int getHeight() const { return height; }
    
    private:
    	TArray<TArray<TArray<T>>> matrix;
    	const unsigned int length;
    	const unsigned int width;
    	const unsigned int height;
    };