如何使用MPI_Sendrecv进行矩阵转置？

#include <stdio.h>
#include <mpi.h>

#define N 4

int main(int argc, char** argv) {
    int rank, size;
    int A[N][N], B[N][N];
    MPI_Status status;

    MPI_Init(&argc, &argv);
    MPI_Comm_rank(MPI_COMM_WORLD, &rank);
    MPI_Comm_size(MPI_COMM_WORLD, &size);

    // 初始化矩阵A
    for (int i = 0; i < N; i++) {
        for (int j = 0; j < N; j++) {
            A[i][j] = i * N + j;
        }
    }

    // 使用MPI_Sendrecv进行矩阵转置
    for (int i = 0; i < N; i++) {
        MPI_Sendrecv(&A[i][0], N, MPI_INT, rank, 0, &B[0][i], N, MPI_INT, rank, 0, MPI_COMM_WORLD, &status);
    }

    // 打印转置后的矩阵B
    if (rank == 0) {
        printf("转置后的矩阵B：\n");
        for (int i = 0; i < N; i++) {
            for (int j = 0; j < N; j++) {
                printf("%d ", B[i][j]);
            }
            printf("\n");
        }
    }

    MPI_Finalize();
    return 0;
}