1 比 0 多的子数组个数

/*
题目思路：
1. 使用流处理数组。
2. 将数组转换为1和-1（0变为-1）。
3. 使用前缀和记录当前的1和-1数量差。
4. 使用一个哈希表来记录当前差值出现的次数。
5. 最后累加满足条件的子数组个数。
*/
import java.util.HashMap;
import java.util.stream.IntStream;
 
public class SubarrayCountStream {
    public int subarraysWithMoreOnesThanZeros(int[] nums) {
        HashMap<Integer, Integer> map = new HashMap<>();
        map.put(0, 1);
        int[] diff = {0};
        return IntStream.of(nums).map(num -> num == 1 ? 1 : -1).reduce(0, (count, num) -> {
            diff[0] += num;
            count += map.getOrDefault(diff[0], 0);
            map.put(diff[0], map.getOrDefault(diff[0], 0) + 1);
            return count;
        });
    }
 
    public static void main(String[] args) {
        SubarrayCountStream scs = new SubarrayCountStream();
        int[] nums = {1, 0, 1, 0, 1};
        System.out.println(scs.subarraysWithMoreOnesThanZeros(nums)); // Output: 9
    }
}