孤独像素 II

/*
 * 题目思路:
 * 我们需要找到所有黑色像素'B'的列数, 使得满足条件的列数的数量等于N
 * 而且每一行只包含恰好N个黑色像素。
 * 
 * 为了实现这一点，我们需要遍历二维字符数组并计数每行和每列中的黑色像素。
 * 我们还需要检查这些行是否有相同的黑色像素列。
 * 
 * 使用Java Stream API来实现上述功能。
 */
 
import java.util.*;
import java.util.stream.*;
 
public class LonelyPixelIIStream {
    public int findBlackPixel(char[][] picture, int N) {
        int rows = picture.length;
        int cols = picture[0].length;
        int[] rowCount = new int[rows];
        int[] colCount = new int[cols];
        Map<String, Long> rowMap = Arrays.stream(picture)
                                          .map(String::new)
                                          .collect(Collectors.groupingBy(row -> row, Collectors.counting()));
 
        IntStream.range(0, rows).forEach(i -> {
            IntStream.range(0, cols).forEach(j -> {
                if (picture[i][j] == 'B') {
                    rowCount[i]++;
                    colCount[j]++;
                }
            });
        });
 
        return (int) IntStream.range(0, rows).filter(i -> rowCount[i] == N)
                               .boxed()
                               .flatMap(i -> IntStream.range(0, cols)
                                                      .filter(j -> picture[i][j] == 'B' && colCount[j] == N)
                                                      .mapToObj(j -> new AbstractMap.SimpleEntry<>(i, j)))
                               .filter(entry -> rowMap.get(new String(picture[entry.getKey()])) == N)
                               .count();
    }
}