Java实现:N个线程并行对数组求和

题目

给定一个数组 int array[],如何用N个线程进行求和。

Java实现

  1. 用 ExecutorService executorService = Executors.newFixedThreadPool(k) 创建k个线程,
  2. 然后,将数组分成k分给每个线程执行;
  3. 使用ExecutorService 的 shutdown() 方法,关闭线程池,不再接受新的任务,但是已经提交的任务会继续执行,直到完成。一旦调用了 shutdown() 方法,如果之后尝试提交新任务给 ExecutorService,则会抛出 RejectedExecutionException 异常。

  4. 最后用 ExecutorService 判定是否执行完 isTerminated()

public static void main(String[] args) {
    // 初始化数组
    int array[] = new int[100];
    for(int i=0; i< array.length; i++) {
        array[i] = i;
    }
    
    // 多线程分段求和
    int K = 4;
    ExecutorService executorService = Executors.newFixedThreadPool(K);
    AtomicInteger atomicInteger = new AtomicInteger(0);
    int len = array.length / K;
    for(int i=0; i< K; i++) {
        int start = i * len;
        int end = i == K - 1 ? array.length : start + len;
        executorService.submit(() -> {
            int sum = 0;
            for(int j = start; j < end; j++) {
                sum += j;
            }
            atomicInteger.addAndGet(sum);
        });
    }
    // 关闭线程池,shutdown执行后,新任务不能加入,老线程任务继续执行完
    executorService.shutdown();
    while(!executorService.isTerminated()) {
        // wait
    }
    // 输出结果
    System.out.println(atomicInteger.get());
}

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值