线程工具类之ForkJoin

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{
         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));
        }
}

你可能感兴趣的:(线程工具类之ForkJoin)