编写一个示例程序,展示如何使用 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框架将数组求和的任务分割成若干个小任务并行执行,最后将每个小任务的结果合并得到总和。

发表评论

后才能评论