MPI收集字符串数组

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

int main(int argc, char** argv) {
    int rank, size;
    MPI_Init(&argc, &argv);
    MPI_Comm_rank(MPI_COMM_WORLD, &rank);
    MPI_Comm_size(MPI_COMM_WORLD, &size);

    // 模拟每个进程中的字符串数组
    char* strings[] = {"Hello", "World", "MPI"};
    int num_strings = sizeof(strings) / sizeof(strings[0]);

    // 分发数组长度
    int* recv_counts = (int*)malloc(sizeof(int) * size);
    MPI_Gather(&num_strings, 1, MPI_INT, recv_counts, 1, MPI_INT, 0, MPI_COMM_WORLD);

    if (rank == 0) {
        // 计算总的字符串数量
        int total_strings = 0;
        for (int i = 0; i < size; i++) {
            total_strings += recv_counts[i];
        }

        // 分配内存
        char** recv_strings = (char**)malloc(sizeof(char*) * total_strings);
        int* displs = (int*)malloc(sizeof(int) * size);
        int displacement = 0;

        // 收集字符串
        MPI_Gatherv(strings, num_strings, MPI_CHAR, recv_strings, recv_counts, displs, MPI_CHAR, 0, MPI_COMM_WORLD);

        // 打印结果
        printf("Collected strings: ");
        for (int i = 0; i < total_strings; i++) {
            printf("%s ", recv_strings[i]);
        }
        printf("\n");

        // 释放内存
        free(recv_strings);
        free(displs);
    }

    free(recv_counts);
    MPI_Finalize();

    return 0;
}