MPI: изменение ячейки матрицы в зависимости от окружающих ее ячеек

#define _CRT_SECURE_NO_WARNINGS
#include "mpi.h"
#include <stdio.h>
#include <stdlib.h>
 
using namespace std;
FILE* startfile;
FILE* finfile;
#define inf 999
 
int* readfromfile(int n, int* arrayA, int* arrayB, int r)
{
    if (r == 1)
        startfile = fopen("matrix58.txt", "r");
    if (r == 2)
        startfile = fopen("matrix148.txt", "r");
    if (r == 3)
        startfile = fopen("matrix298.txt", "r");
 
    int i = 0;
    while (fscanf(startfile, "%d", &arrayA[i]) != EOF)
    {
 
        printf("%d  ", arrayA[i]);
        i++;
    }
    for (int i = 0; i < (n + 2) * (n + 2); i++)
    {
        arrayB[i] = arrayA[i];
        printf("%d  ", arrayB[i]);
    }
    fclose(startfile);
    printf(" process 0 send create buffer \n ");
    return arrayA, arrayB;
}
 
int main(int argc, char* argv[])
{
    system("chcp 1251");
    system("cls");
    int n, rank, size;
 
    int* sendA, * sendB, * finalC, * recA, * recC;
 
    char processor_name[MPI_MAX_PROCESSOR_NAME];
 
    MPI_Init(&argc, &argv);
    int h;
    MPI_Comm_size(MPI_COMM_WORLD, &size); // max number of processors
    MPI_Comm_rank(MPI_COMM_WORLD, &rank); // our rank
 
    printf("Введите h: \n");
    scanf("%d", &h);
    //h = atoi(argv[1]);
    int r;// = atoi(argv[2]);
 
    printf("Если н = 50, введите 1\nЕсли н = 100, введите 2\nЕсли н = 500, введите 3\n");
    scanf("%d", &r);
 
    if (r == 1)
    {
        n = 60;
    }
    else if (r == 2)
    {
        n = 150;
    }
    else if (r == 3)
    {
        n = 300;
    }
    else
    {
        printf("Введено неверное значение\n");
        return 0;
    }
 
    printf("rows is %d \n", n);
    MPI_Status status;
    MPI_Comm comm1, comm2;
 
    sendA = (int*)malloc(n * n * sizeof(int*));
    sendB = (int*)malloc(n * n * sizeof(int*));
    finalC = (int*)malloc(n * n * sizeof(int*));
 
 
    // p0 is reading the file
    if (rank == 0) {
        sendA, sendB = readfromfile(n, sendA, sendB, r);
    }
 
    MPI_Barrier(MPI_COMM_WORLD); /* barrier is needed if no necessary synchronization for the timing is ensured yet */
    double start = MPI_Wtime(); /* only 1 process needs to do this */
 
        // first broad cast the n to all others
    MPI_Bcast(&n, 1, MPI_INT, 0, MPI_COMM_WORLD);
    MPI_Bcast(&h, 1, MPI_INT, 0, MPI_COMM_WORLD);
    printf("  n / p is %d \n ", n / size);
 
 
    // allocating lical data
    recA = (int*)malloc(n / size * n * sizeof(int*));
    if (sendB == NULL) {
        printf("NULL\n");
        sendB = (int*)malloc(n * n * sizeof(int*));
    }
    recC = (int*)malloc(n / size * n * sizeof(int*));
    printf(" process %d create rec buffer \n ", rank);
 
 
    // now p0 scatter matrix sendA to all 
    MPI_Scatter(sendA, n * n / size, MPI_INT, recA, n * n / size, MPI_INT, 0, MPI_COMM_WORLD);
    for (int i = 0; i < (n / size); i++)
        printf(" process %d sendA [%d] = % d \n ", rank, i, recA[i]);
 
 
    //now p0 broadcast sendB to all others
    MPI_Bcast(sendB, n * n, MPI_INT, 0, MPI_COMM_WORLD);
 
    for (int i = 0; i < (n * n); i++)
        printf(" process %d sendB [%d] = % d \n ", rank, i, sendB[i]);
 
 
    // all do this part to calculate recC as multiplicated matrix       
    for (int i = 0; i < n; i++)
    {
        for (int j = 0; j < n; j++)
        {
 
            if (sendA[i * n + j] != -inf)
            {
                if ((sendA[i * n + j] - sendA[i * (n - 1) + j]) < h)//вверх
                    recC[i * n + j] = sendA[i * n + j] - 3;
                if ((sendA[i * n + j] - sendA[i * n + (j - 1)]) < h)//слева
                    recC[i * n + j] = sendA[i * n + j] - 7;
                if ((sendA[i * n + j] - sendA[i * (n + 1) + j]) < h)//низ
                    recC[i * n + j] = sendA[i * n + j] - 5;
                if ((sendA[i * n + j] - sendA[i * n + (j + 1)]) < h)//справа
                    recC[i * n + j] = sendA[i * n + j] - 6;
                else
                    recC[i * n + j] = sendA[i * n + j];
            }
            else
            {
                recC[i * n + j] = -inf;
            }
            printf(" process %d recC [%d] = % d \n", rank, i * n + j, recC[i * n + j]);
 
        }
    }
 
    // nopw p0 will gather all result data from all prcesses
    MPI_Gather(recC, n * n / size, MPI_INT, finalC, n * n / size, MPI_INT, 0, MPI_COMM_WORLD);
 
 
    // here is the last point of calculating the time       
    MPI_Barrier(MPI_COMM_WORLD); /* barrier is needed if no necessary synchronization for the timing is ensured yet */
    double end = MPI_Wtime(); /* only 1 process needs to do this */
 
    // we write the time
    double time = end - start;
    printf(" ********** process %d : time = % d \n", rank, time);
 
 
    // here p0 is writing the output into the file  
    if (rank == 0) {
 
        FILE* out = fopen("matrix_result.txt", "w+");
        if (out == NULL)
        {
            printf("Error opening file!\n");
            exit(1);
        }
 
        for (int i = 0; i < n; i++)
        {
            for (int j = 0; j < n; j++)
                fprintf(out, " %d ", finalC[i * n + j]);
            fprintf(out, "\n");
        }
        fclose(out);
    }
 
    MPI_Finalize();
    printf("Process: %d, size: %d\n", rank, size);
    return 0;
}