将嵌套的“for 循环”转换为模块

如何解决将嵌套的“for 循环”转换为模块

我有一段完整的 C++ 代码。我已经成功地模块化了现有的功能。我的任务要求对 main() 的某些部分进行模块化。在这种情况下，我希望将一个嵌套的“for 循环”变成一个模块。它有一个对退出模块的函数调用。我需要新模块返回一个数组“number_of_ants”。这样做的最佳方法是什么？

#include <iostream>
#include "linear.h"
int main(){
int length     = 70;
int* number_of_ants = new int[length*length]; 
int total_ants = 40000;

   for (int i = 0; i < length; i++) {
        for (int j = 0; j < length; j++) {
        int n = linear(i,j,length,length);
        number_of_ants[n] = ((long long)(n+1)*total_ants)/(length*length) - ((long long) 
                            (n)*total_ants)/(length*length);
        }
   }
}

我需要返回 number_of_ants[] 这是一个数组。 注意：函数 linear() 只是将自然为 2D（行、列）的数据转化为向量。