Java8:如何基于Iterable参数用Lambda表达式获取最接近给定值的n个数
解决方法:实现返回最接近给定值的n个数字的Iterable
嘿,这个问题用Java的Stream API就能很简洁地搞定,而且完全不会修改原参数。我给你两种版本的实现,你可以根据对返回结果顺序的需求来选择:
基础版本(按接近程度排序)
这个版本会把最接近给定值的数字放在最前面,依次往后排列:
import java.util.Comparator; import java.util.NoSuchElementException; import java.util.stream.Collectors; import java.util.stream.StreamSupport; public Iterable<Integer> findClosestNumbers(int givenValue, Iterable<Integer> numbers, int n) { // 校验输入的Iterable是否为空,为空则抛出指定异常(和你提供的示例逻辑保持一致) if (!numbers.iterator().hasNext()) { throw new NoSuchElementException("No value present"); } // 将Iterable转为Stream,按与givenValue的绝对值差排序,取前n个转成List(List本身属于Iterable) return StreamSupport.stream(numbers.spliterator(), false) .sorted(Comparator.comparingInt(num -> Math.abs(num - givenValue))) .limit(n) .collect(Collectors.toList()); }
匹配示例顺序的版本(按数字升序排列)
如果你需要像示例那样,返回的结果按数字本身的升序排列(比如givenValue=6、n=2时返回{4,5}),可以在排序时先按接近程度,再补充数字大小的排序规则:
import java.util.Comparator; import java.util.NoSuchElementException; import java.util.stream.Collectors; import java.util.stream.StreamSupport; public Iterable<Integer> findClosestNumbers(int givenValue, Iterable<Integer> numbers, int n) { if (!numbers.iterator().hasNext()) { throw new NoSuchElementException("No value present"); } return StreamSupport.stream(numbers.spliterator(), false) // 先按与给定值的绝对值差排序,再按数字本身升序排序 .sorted(Comparator.comparingInt((Integer num) -> Math.abs(num - givenValue)) .thenComparingInt(num -> num)) .limit(n) .collect(Collectors.toList()); }
核心思路说明
- Iterable转Stream:用
StreamSupport.stream()把Iterable转换成Stream,这是处理Iterable最便捷的方式之一,不需要额外的集合转换操作。 - 排序逻辑:通过
Comparator.comparingInt()直接基于与givenValue的绝对值差排序,逻辑清晰直观。如果需要调整返回顺序,只需要补充额外的排序规则即可。 - 边界处理:开头的空校验和你提供的示例保持一致,确保输入为空时抛出指定异常。
- 无参数修改:整个过程完全没有改动传入的
numbersIterable,完全符合要求。
内容的提问来源于stack exchange,提问作者Rosso
相关产品推荐
相关产品推荐

