通过流将字符串数组转换为整数数组

Question

我想将给定字符串数组中每个字符串中的 '1' 字符的数量计算出来。

示例：

输入：["00101010", "10010", "11111"]
输出：[3, 2, 5]

String []b = ["00101010", "10010", "11111"];
int [] c = Arrays.stream(b).mapToInt(r -> countChar(r, '1')).toArray();

int countChar(String string, char x) {
    return (int)string.chars().filter(ch -> ch == x).count();
}

如何将这段代码写成一个单行表达式？

我尝试将 countChar(r, '1') 替换为 (int)string.chars().filter(ch -> ch == x).count() 及其各种组合，但错误信息太多让我无法理解。

Reimeus · Answer

如果数组中的所有字符都是 0 和 1，那么类似下面的方法应该可行。通过移除所有零字符，字符串的长度即为 '1' 的个数。但这不是最高效的方法，因为 replace 方法会创建一个临时字符串。

.stream()
.map(s -> s.replace("0", ""))
.mapToInt(String::length)
.toArray();

如果你有其他非 '1' 字符，可以创建一个正则表达式来替换除 '1' 之外的所有字符为空字符串。

bdukes · Answer

你可以将 countChar(r, '1') 方法调用替换为其方法体：

int[] c = Arrays.stream(b)
                .mapToInt(r -> (int) r.chars()
                                      .filter(ch -> ch == '1')
                                      .count())
                .toArray();

Russ Cam · Answer

其他答案已经很好了。但是，我有两点建议：(a)，在处理单个字符时，建议养成使用码点整数的习惯，而不是使用char类型；(b)，你在评论中提到你希望进行并行编程。

数字“1”的码点是49。

    int[] counts =
        Stream
        .of(  new String[] { "00101010", "10010", "11111" } )
        .parallel() // 使用并行流
        .mapToLong( ( String s ) -> s.codePoints().filter( (int codePoint) -> codePoint == 49 ).count() )
        .mapToInt( Math :: toIntExact )
        .toArray() ;

请注意，并行并不一定意味着更快。

可以在Ideone.com上查看这段代码的运行情况。