情感丰富的文字

/*
 * 思路：
 * 1. 使用 Java Stream 来简化代码逻辑。
 * 2. 仍然使用 isStretchy 函数判断每个查询单词是否可扩展。
 */

import java.util.stream.Stream;

public class ExpressiveWordsStream {
    public static void main(String[] args) {
        String s = "heeellooo";
        String[] words = {"hello", "hi", "helo"};
        System.out.println(expressiveWords(s, words)); // 输出: 1
    }

    public static long expressiveWords(String s, String[] words) {
        return Stream.of(words).filter(word -> isStretchy(s, word)).count();
    }

    public static boolean isStretchy(String s, String word) {
        int i = 0, j = 0;
        while (i < s.length() && j < word.length()) {
            if (s.charAt(i) != word.charAt(j)) return false;
            int lenS = getRepeatedLength(s, i);
            int lenW = getRepeatedLength(word, j);
            if (lenS < 3 && lenS != lenW || lenS >= 3 && lenS < lenW) return false;
            i += lenS;
            j += lenW;
        }
        return i == s.length() && j == word.length();
    }

    private static int getRepeatedLength(String s, int index) {
        int length = 0;
        char c = s.charAt(index);
        while (index + length < s.length() && s.charAt(index + length) == c) {
            length++;
        }
        return length;
    }
}