ForkJoin 是JAVA并发编程提供的一个工具类,可以让我们不需要了解Runnable、Thread知识,只要遵循forkjoin开发模式,就可以写出很好的多线程并发程序。
ForkJoin是一种分而治之思想,什么是分而治之呢?
对于一个规模为n的问题,若该问题可以很容易的解决则直接解决,若不行,就分成k个较小的子问题,这些子问题相互独立且与原问题形式相同,递归的解决这些子问题,然后将结果合并得到原问题的解,这就是分而治之。下面使用forkjoin实现一个归并排序算法
public class MergeSort{
//合并结果
public static int[] merge(int[] left,int[] right){
int result = new int[left.length + right.length];
for(int index = 0,i = 0,j = 0; index < result.length; index++){
if(i >= left.length){
result[index] = right[j++];
}else if(j >= right.length){
result[index] = left[i++];
}else if(left[i] > right[i]){
result[index] = right[j++];
}else{
result[index] = left[i++];
}
}
}
public static int[] sort(int[] array){
if(array.length == 0){
return array;
}
//当前待排序数据,该元素之前的元素均已排序过
int currentValue;
for(int i = 0;i < array.length - 1;i++){
int preIndex = i;
currentValue = array[preIndex + 1];
/*在已被排序过数据中倒序寻找合适的位置,如果当前待排序数据比待比较的数据小,则将标记的元素向后移一位*/
while(preIndex > 0 && currentValue < array[preIndex]){
array[preIndex + 1] = array[preIndex];
preIndex--;
}
array[preIndex + 1] = currentValue;
}
return array;
}
}
public class FKsort{
public static class SumTask extends RecursiveTask<int[]>{
private int[] src;
//这里定义两个是两两进行比较,是最小单元
private static final int THEHOULD = 2;
public SumTask(int[] src){
this.src = src;
}
@Override
protected int[] compute(){
if(src.length <= THEHOULD){
return MergeSort.sort(array);
}else{
int mid = array.length / 2;
//复制左边数组
SumTask left = new SumTask(Arrays.copyOfRange(src,0,mid));
//复制右边数组
SumTask right = new SumTask(Arrays.copyOfRange(src,mid,src.length));
invokeAll(left,right);
int[] leftResult = left.join();
int[] rightResult = right.join();
return MergeSort.merge(leftResult,rightResult);
}
}
}
public static void main(String[] args){
ForkJoinPool pool = new ForkJoinPool();
//这里可以定制一个数组
int[] array = {3,4,5,1,2,0,8,10,222,3456};
SumTask task = new SumTask(array);
//执行命令
pool.invoke(task);
//获取结果
int[] newArray = task.join();
System.out.println(Arrays.toString(newArray));
}
}
网友评论