孤独像素 I

/*
 * 思路:
 * 1. 使用流操作来计算每行和每列'B'的出现次数
 * 2. 再次使用流操作来检查每个'B'是否是孤独像素
 */
import java.util.Arrays;
import java.util.stream.IntStream;
 
public class LonelyPixelIStream {
    public int findLonelyPixel(char[][] picture) {
        int m = picture.length;
        int n = picture[0].length;
        int[] rowCount = new int[m];
        int[] colCount = new int[n];
        // 记录每行和每列'B'的出现次数
        IntStream.range(0, m).forEach(i -> 
            IntStream.range(0, n).forEach(j -> {
                if (picture[i][j] == 'B') {
                    rowCount[i]++;
                    colCount[j]++;
                }
            })
        );
        // 计算孤独像素的数量
        return (int) IntStream.range(0, m).flatMap(i -> 
            IntStream.range(0, n).filter(j -> 
                picture[i][j] == 'B' && rowCount[i] == 1 && colCount[j] == 1
            )
        ).count();
    }
}