最美子字符串的数目

/*
 * 思路：
 * 1. 利用Java Stream API进行流式处理。
 * 2. 和普通Java代码思路一样，只是在处理过程中使用Stream进行处理。
 */

import java.util.HashMap;
import java.util.stream.IntStream;

public class BeautifulSubstringsStream {
    public int countBeautifulSubstrings(String word) {
        int[] prefixOddEven = new int[10];
        HashMap<Integer, Integer> map = new HashMap<>();
        final int[] currentState = {0};
        final int[] result = {0};

        map.put(0, 1); // 初始化前缀状态0的出现次数

        IntStream.range(0, word.length()).forEach(i -> {
            int index = word.charAt(i) - 'a';
            prefixOddEven[index] ^= 1; // 更新前缀状态
            currentState[0] = 0;
            for (int j = 0; j < 10; j++) {
                currentState[0] |= (prefixOddEven[j] << j); // 计算当前前缀状态
            }
            result[0] += map.getOrDefault(currentState[0], 0);
            map.put(currentState[0], map.getOrDefault(currentState[0], 0) + 1);
        });
        return result[0];
    }

    public static void main(String[] args) {
        BeautifulSubstringsStream bss = new BeautifulSubstringsStream();
        System.out.println(bss.countBeautifulSubstrings("aba")); // 4
        System.out.println(bss.countBeautifulSubstrings("aabb")); // 9
        System.out.println(bss.countBeautifulSubstrings("he")); // 2
    }
}