找出数组中的美丽下标 I

/*
题目思路：
1. 使用 Java Stream API 处理字符串 s。
2. 找到所有的 a 的出现位置。
3. 找到所有的 b 的出现位置。
4. 对每个 a 的出现位置 i，检查是否存在一个 b 的位置 j 使得 |j - i| <= k。
5. 如果满足条件，将 i 添加到美丽下标列表中。
6. 最后对美丽下标列表进行排序并返回。
*/
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.stream.Collectors;
import java.util.stream.IntStream;

public class BeautifulIndexesStream {
    public static List<Integer> findBeautifulIndexes(String s, String a, String b, int k) {
        int aLen = a.length();
        int bLen = b.length();

        List<Integer> aIndexes = IntStream.range(0, s.length() - aLen + 1)
                .filter(i -> s.substring(i, i + aLen).equals(a))
                .boxed()
                .collect(Collectors.toList());

        List<Integer> bIndexes = IntStream.range(0, s.length() - bLen + 1)
                .filter(i -> s.substring(i, i + bLen).equals(b))
                .boxed()
                .collect(Collectors.toList());

        List<Integer> result = aIndexes.stream()
                .filter(i -> bIndexes.stream().anyMatch(j -> Math.abs(j - i) <= k))
                .sorted()
                .collect(Collectors.toList());

        return result;
    }

    public static void main(String[] args) {
        String s = "isawsquirrelnearmysquirrelhouseohmy";
        String a = "my";
        String b = "squirrel";
        int k = 15;
        System.out.println(findBeautifulIndexes(s, a, b, k)); // 输出：[16, 33]
    }
}