编写一个示例程序,展示如何使用 Fork/Join 框架进行并行计算。
以下是使用Fork/Join框架实现一个并行计算数组元素之和的示例:
import java.util.concurrent.RecursiveTask;
import java.util.concurrent.ForkJoinPool;
public class SumArray {
private static class SumTask extends RecursiveTask<Integer> {
private static final int THRESHOLD = 50; // 设定一个阈值
private final int[] array;
private final int start;
private final int end;
public SumTask(int[] array, int start, int end) {
this.array = array;
this.start = start;
this.end = end;
}
@Override
protected Integer compute() {
if (end - start <= THRESHOLD) {
// 如果任务足够小,直接计算
int sum = 0;
for (int i = start; i < end; i++) {
sum += array[i];
}
return sum;
} else {
// 如果任务大于阈值,继续分割
int mid = (start + end) / 2;
SumTask left = new SumTask(array, start, mid);
SumTask right = new SumTask(array, mid, end);
left.fork();
right.fork();
return left.join() + right.join();
}
}
}
public static void main(String[] args) {
int[] array = new int[1000];
for (int i = 0; i < array.length; i++) {
array[i] = i;
}
ForkJoinPool pool = new ForkJoinPool();
SumTask task = new SumTask(array, 0, array.length);
Integer result = pool.invoke(task);
System.out.println("The sum is: " + result);
}
}
这个示例中,我们首先创建了一个大小为1000的数组,然后使用Fork/Join框架将数组求和的任务分割成若干个小任务并行执行,最后将每个小任务的结果合并得到总和。