如何解决在使用动态 2D 数组进行矩阵乘法的 MPI 进行并行编程时的问题

我正在尝试使用 MPI 创建三个矩阵 a，b，c，其中 c = a*b。此外，我将这些矩阵的长度作为 N(对所有矩阵通用(，因为我必须创建一个方阵。但是，每当我在运行时输入 N 的值时，我都会收到分段错误错误，如果我在程序中输入 N 的值，那么它就可以正常工作。

我已经尝试过分散和收集，如下所示：使用 Mpi_Scatter 和 Mpi_Gather 的矩阵乘法.现在我必须动态地处理，以便可以检查程序执行它所消耗的时间。只是想通知我已经完成了这个OpenMP，这很棒，但想比较哪一个真的很好，即OpenMP或MPI。

#include <iostream>
#include <math.h>
#include <sys/time.h>
#include <stdlib.h>
#include <stddef.h>
#include "mpi.h"

int main(int argc, char *argv[])
{
    int i, j, k, rank, size, tag = 99, blksz, sum = 0,N=0;
    MPI_Init(&argc, &argv);
    MPI_Comm_size(MPI_COMM_WORLD, &size);
    MPI_Comm_rank(MPI_COMM_WORLD, &rank);
    int aa[N],cc[N];
    if(rank ==0)
    {
      std::cout << "input value of N" << 'n';
      std::cin >> N;
    }
    MPI_Bcast(&N, 1, MPI_INT, 0, MPI_COMM_WORLD);
    int **a = new int*[N];
    for (int i = 0; i < N; i++)
        a[i] = new int[N];
    int **b = new int*[N];
    for (int i = 0; i < N; i++)
        b[i] = new int[N];
    int **c = new int*[N];
    for (int i = 0; i < N; i++)
        c[i] = new int[N];
        if (rank == 0)
        {
            for (int i = 0; i < N; i++)
            {
                for (int j = 0; j < N; j++)
                {
                        a[i][j] =rand() % 10;
                        std::cout << a[i][j];
                }
                std::cout << 'n';
            }
            std::cout << 'n';
            for (int i = 0; i < N; i++)
            {
                for (int j = 0; j < N; j++)
                {
                        b[i][j] =rand() % 10;
                        std::cout << b[i][j];
                }
                std::cout << 'n';
            }
        }
    MPI_Scatter(a, N*N/size, MPI_INT, aa, N*N/size, MPI_INT,0,MPI_COMM_WORLD);
    //broadcast second matrix to all processes
    MPI_Bcast(b, N*N, MPI_INT, 0, MPI_COMM_WORLD);
    MPI_Barrier(MPI_COMM_WORLD);
          //perform vector multiplication by all processes
          for (i = 0; i < N; i++)
          {
                    for (j = 0; j < N; j++)
                    {
                            sum = sum + aa[j] * b[j][i];  //MISTAKE_WAS_HERE
                    }
                    cc[i] = sum;
                    sum = 0;
            }
    MPI_Gather(cc, N*N/size, MPI_INT, c, N*N/size, MPI_INT, 0, MPI_COMM_WORLD);
    MPI_Barrier(MPI_COMM_WORLD);
    MPI_Finalize();

    if (rank == 0)                         //I_ADDED_THIS
    {
      for (i = 0; i < N; i++) {
              for (j = 0; j < N; j++)
              {
                      std::cout << a[i][j]<< 'n';
              }
              std::cout << 'n';
      }
      std::cout << 'n' << 'n';
    }
    delete *a;
    delete *b;
    delete *c;
}

我得到的错误是：

mpirun 注意到节点本地主机上具有 PID 3580 的进程等级 3 在信号 11 上退出(分段错误(。

我只是想在这里完成矩阵乘法。