如何在C++中使用MPI实现简单循环的负载均衡

#include <mpi.h>

int main(int argc, char** argv) {
    MPI_Init(&argc, &argv);
    int numProcesses, rank;
    MPI_Comm_size(MPI_COMM_WORLD, &numProcesses);
    MPI_Comm_rank(MPI_COMM_WORLD, &rank);
    // ...
}

int numTasks = 100; // 循环任务的总数
int tasksPerProcess = numTasks / numProcesses; // 每个进程处理的任务数量
int startTask = rank * tasksPerProcess; // 当前进程开始处理的任务编号
int endTask = startTask + tasksPerProcess; // 当前进程结束处理的任务编号
if (rank == numProcesses - 1) {
    // 最后一个进程处理剩余的任务
    endTask = numTasks;
}

for (int i = startTask; i < endTask; i++) {
    // 执行循环任务
    // ...
}

MPI_Barrier(MPI_COMM_WORLD);

// 假设每个进程都有一个局部结果localResult
int globalResult;
MPI_Reduce(&localResult, &globalResult, 1, MPI_INT, MPI_SUM, 0, MPI_COMM_WORLD);

MPI_Finalize();