BitSet.stream获取所有置位变量索引的流

在Ja va的BitSet操作中，高效地遍历所有被设置为“1”的位，是日常开发里一个很实际的需求。过去我们可能习惯用循环配合nextSetBit()，但现在有了更现代、更符合函数式编程风格的选择——BitSet.stream()。简单来说，前者适合需要精细控制遍历过程的场景，而后者则让链式处理和集合操作变得异常流畅。

BitSet.stream() 直接返回所有置位索引的升序 IntStream，时间复杂度 O(k)，支持链式操作；而 nextSetBit() 适用于需控制起点或中途跳出的场景，遍历中修改 BitSet 时更可控。

想要一次性拿到所有值为 true 的位索引？现在不用再写手动循环了。直接调用 BitSet.stream()，它会返回一个按升序排列的 IntStream。即便是空的BitSet，它也会安全地返回一个空流，既简洁又高效。

stream() 的核心特点

这个方法的设计很聪明，它并不会傻乎乎地去扫描底层整个long数组。其内部做了优化，只遍历那些真正被设置过的位，因此时间复杂度大致是 O(k)，这里的 k 就是实际置位的数量。返回的流是延迟求值的，这意味着你可以轻松地进行各种链式操作。比如：

• bs.stream().filter(i -> i % 2 == 0).forEach(System.out::println); —— 只筛选并处理偶数索引。
• int[] indices = bs.stream().toArray(); —— 直接把所有索引收集成一个数组。
• long count = bs.stream().count(); —— 快速统计置位个数，这效果和调用 cardinality() 方法是一样的。

和 nextSetBit() 的区别

那么，什么时候该用stream()，什么时候又该用传统的nextSetBit(int fromIndex)呢？关键在于场景。

stream() 天生适合函数式风格和需要组合多个操作的场合。如果你想从第100位之后开始寻找第一个置位，那么nextSetBit(100)显然是更直接的选择。反过来，如果你的需求是把所有置位索引转成一个List，顺便去个重、排个序，那么一行stream().boxed().collect(Collectors.toList())写起来就自然多了。

还有一个重要区别在于可变性。如果在遍历过程中，你有可能需要修改BitSet本身的内容，那么nextSetBit()在循环里会更可控。因为stream()返回的流可以看作是一个快照，它不会反映遍历开始后对BitSet所做的修改。

常见误用提醒

这里有个常见的“踩坑”写法需要警惕：不要试图用length()或size()配合循环get(i)来模拟流式遍历。

• 首先，length()返回的是“最高置位索引 + 1”，而不是BitSet的总容量（size()返回的是底层数组的位数）。这意味着中间可能充斥着大量未设置的false位，循环效率很低。
• 其次，get(i)在索引越界时会安静地返回false，而不是抛出异常，这很容易导致静默的逻辑错误。
• 最后，这种逐个判断的方式，其时间复杂度是O(n)，而stream()和nextSetBit()都是O(k)，在稀疏位集（即置位很少）的情况下，性能差异会非常明显。

简单示例

看一个具体的例子就一目了然了。下面的代码设置了第3、7、15位，然后使用stream()将它们打印出来：

BitSet bs = new BitSet();
bs.set(3); bs.set(7); bs.set(15);
bs.stream().forEach(i -> System.out.print(i + " "));
// 输出：3 7 15