重载运算符<<用于查找二维数组的总和

Question

我正在用 C++ 做一些练习题。我遇到了一个问题，我想找到 2d 数组 的元素总和。我可以编写一个返回总和的 get sum 方法。但我正在探索是否可以重载“operator<<”方法以达到相同的结果。

#include <iostream>
using namespace std;

int operator<<(const int arr[5][5])
{
   int sum = 0;
   for (int i = 0; i < 5; i++)
   {
      for (int j = 0; j < 5; j++)
      {
         sum += arr[i][j];
      }
   }
   return sum;
}

int main()
{
   int arr[5][5] = { {1,2,3,4,5},
                    {2,3,4,5,6},
                    {3,4,5,6,7},
                    {4,5,6,7,8},
                    {5,6,7,8,9} };
   cout << &arr << endl;
}

我想像std::cout 方法一样获得总和。这可能吗？

Answer 1

您可以通过为operator<< 提供模板重载来获得所需的输出，该模板采用const& int[row][col]，如下所示。 这仅适用于int[row][col]。

(See Live Online)

#include <iostream>
#include <numeric> // std::accumulate

template<std::size_t M, std::size_t N> 
std::ostream& operator<<(std::ostream& out, const int (&arr)[M][N]) /* noexcept */
{
    int sum = 0;
    for (std::size_t i = 0; i < M; ++i)
    {
#if false // either using `std::accumulate`        
        sum += std::accumulate(arr[i], arr[i] + N, 0);

#elif true // or using for- loop
        for (std::size_t j = 0; j < N; j++)
            sum += arr[i][j];
#endif
    }
    return out << sum;
}

---

旁注：

• 但是，您正在更改 'operator
• 同时避免使用using namespace std; 练习。

Answer 2

使用std::accumulate，由于数组中的内存是线性的，所以可以使用数组的开始和结束作为迭代器。

#include <iostream>
#include <numeric>  

int main()
{
    int arr[5][5] = { {1,2,3,4,5},
            {2,3,4,5,6},
            {3,4,5,6,7},
            {4,5,6,7,8},
            {5,6,7,8,9} };
    int sum = 0;
    sum += std::accumulate(arr[0], arr[0]+sizeof(arr)/sizeof(arr[0][0]), 0);
    std::cout << "Sum of array arr[5][5]: " << sum << "\n";


}