统计以给定字符开头和结尾的子字符串总数

/*
 * 思路：
 * 使用Java Stream API来实现同样的逻辑。
 * 首先将字符串s转换成字符流，找到所有字符c出现的索引位置，
 * 对于每一个索引位置i，使用一个嵌套流来计算从i开始到后面所有的以字符c结尾的子字符串数量。
 */
import java.util.stream.IntStream;

public class SolutionStream {
    public long countSubstrings(String s, char c) {
        return IntStream.range(0, s.length())
                .filter(i -> s.charAt(i) == c)
                .mapToLong(i -> IntStream.range(i, s.length())
                        .filter(j -> s.charAt(j) == c)
                        .count())
                .sum();
    }
    public static void main(String[] args) {
        SolutionStream sol = new SolutionStream();
        System.out.println(sol.countSubstrings("abada", 'a')); // 6
        System.out.println(sol.countSubstrings("zzz", 'z'));   // 6
    }
}