字符串的好分割数目

/*
 * 思路：
 * 1. 使用 Java Stream API 进行字符串处理。
 * 2. 统计每个位置左边和右边不同字符的数量。
 * 3. 用两个数组 leftCounts 和 rightCounts 分别记录从左到右和从右到左的不同字符数量。
 * 4. 最后判断 leftCounts[i] 是否等于 rightCounts[i+1]。
 */

import java.util.*;
import java.util.stream.Collectors;

public class GoodSplitsStream {
    public int numSplits(String s) {
        int n = s.length();
        int[] leftCounts = new int[n];
        int[] rightCounts = new int[n];
        Set<Character> leftSet = new HashSet<>();
        Set<Character> rightSet = new HashSet<>();
        
        // 从左到右统计不同字符数
        IntStream.range(0, n).forEach(i -> {
            leftSet.add(s.charAt(i));
            leftCounts[i] = leftSet.size();
        });
        
        // 从右到左统计不同字符数
        IntStream.iterate(n - 1, i -> i >= 0, i -> i - 1).forEach(i -> {
            rightSet.add(s.charAt(i));
            rightCounts[i] = rightSet.size();
        });
        
        // 判断好分割数量
        long goodSplits = IntStream.range(0, n - 1)
                                   .filter(i -> leftCounts[i] == rightCounts[i + 1])
                                   .count();
        
        return (int) goodSplits;
    }
    
    public static void main(String[] args) {
        GoodSplitsStream gss = new GoodSplitsStream();
        System.out.println(gss.numSplits("aacaba")); // 输出: 2
        System.out.println(gss.numSplits("abcd"));   // 输出: 1
        System.out.println(gss.numSplits("aaaaa"));  // 输出: 4
        System.out.println(gss.numSplits("acbadbaada")); // 输出: 2
    }
}