Java 8 使用 Arrays.stream().sum()求和错误的处理 2208. 将数组和减半的最少操作次数

问题背景

今天在力扣签到题时,使用数组流.sum()函数对所有内容求和,但最终因为求和方法的不同,导致我 算法在某些情况下失败了,当测试用例较小时,可以通过,当数值较大时会出现错误

  1. 将数组和减半的最少操作次数
    给你一个正整数数组 nums 。每一次操作中,你可以从 nums 中选择 任意 一个数并将它减小到 恰好 一半。(注意,在后续操作中你可以对减半过的数继续执行操作)
    请你返回将 nums 数组和 至少 减少一半的 最少 操作数。

示例 1:
输入:nums = [5,19,8,1]
输出:3
解释:初始 nums 的和为 5 + 19 + 8 + 1 = 33 。
以下是将数组和减少至少一半的一种方法:
选择数字 19 并减小为 9.5 。
选择数字 9.5 并减小为 4.75 。
选择数字 8 并减小为 4 。
最终数组为 [5, 4.75, 4, 1] ,和为 5 + 4.75 + 4 + 1 = 14.75 。
nums 的和减小了 33 - 14.75 = 18.25 ,减小的部分超过了初始数组和的一半,18.25 >= 33/2 = 16.5 。
我们需要 3 个操作实现题目要求,所以返回 3 。
可以证明,无法通过少于 3 个操作使数组和减少至少一半。
示例 2:
输入:nums = [3,8,20]
输出:3
解释:初始 nums 的和为 3 + 8 + 20 = 31 。
以下是将数组和减少至少一半的一种方法:
选择数字 20 并减小为 10 。
选择数字 10 并减小为 5 。
选择数字 3 并减小为 1.5 。
最终数组为 [1.5, 8, 5] ,和为 1.5 + 8 + 5 = 14.5 。
nums 的和减小了 31 - 14.5 = 16.5 ,减小的部分超过了初始数组和的一半, 16.5 >= 31/2 = 16.5 。
我们需要 3 个操作实现题目要求,所以返回 3 。
可以证明,无法通过少于 3 个操作使数组和减少至少一半。

该题目较为简单,使用优先队列每次取最大的值作为贪心策略就可以解决
当使用for循环求和时,通过
而使用Arrays.stream().sum()求和时则报错

class Solution {
         public int halveArray(int[] nums) {
      PriorityQueue<Double> pq = new PriorityQueue<Double>((a, b) -> b.compareTo(a));
      		//当使用sum1作为判断条件时,部分测试用例不通过
            double sum1 = Arrays.stream(nums).sum();
            //用sum2作为判断条件时,通过
            double sum2=0.0,t=0.0;
            int count=0;
            for (int num:nums){
                pq.offer((double) num);
                sum2+=num;
            }   
            while(t<sum2/2){
                double x=pq.poll();
                t+=x/2;
                pq.offer(x/2);
                count++;
            }
            return count;
    }
}

问题分析

部分测试通过,但数值较大时不通过,猜测是溢出
查阅资料后发现
Arrays.stream(int[])返回一个IntStream,sum()方法返回int结果,如果超过Integer.Max_VALUE.就会发生溢出

问题解决

将sum结果添加到double总数中,循环就不会溢出,同时需要将IntStream转换为DoubleStream

double sum1=Arrays.Stream(nums).asDoubleStream().sum();

再次运行,通过

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值