首页 > 解决方案 > java - 如何使用Java中的流比较和操作一个列表中的两个相邻元素?

问题描述

背景是我有两个字符串类型变量 str1 和 str2 作为输入。最后,我必须返回一个列表,其中包含 str1 的连续前缀,该前缀小于 str2 中的相关前缀。

我有这样的Java代码:

public List<Character> getPrefix(String str1, String str2) {
    int index = 0;
    List<Character> res = new ArrayList<>();
    //str1 = "1243"
    //str2 = "2324"
    // The answer will be "12".

    while (index < str1.length() && index < str2.length() && str1.charAt(index) <= str2.charAt(index)) {
        res.add(str1.charAt(index));
        index++;
    } 

    return res;
}
//the return type could either be List<String> or List<Character>

我被要求在流中转换此代码而不使用 while 或 for 循环,只是在流方法中。我打算像这样转换这段代码

List<String> list = new ArrayList<>();
list.add(str1);
list.add(str2);
List<String> res = list.stream()
.filter()
.reduce();

我发现filter()方法可以选择与给定谓词匹配的元素,并且reduce()方法可以使用标识和累加器来获得一个最终结果。

但是我发现我既没有办法操作一个列表中的两个相邻元素,也没有办法获得一个指针来比较和遍历列表中每个元素中的每个字符(该元素是字符串类型)。

那么有什么方法可以操作一个列表中的两个相邻元素,以便我可以比较它们在同一位置的字符。

标签: javajava-8java-stream

解决方案


你可以:

  1. 生成索引流
  2. 使用索引获取两个字符串的字符
  3. 有效时选择字符

    // The magic
    public static List<Character> getPrefix(String str1, String str2) {
    
        return IntStream
            .range(0, Math.min(str1.length(), str2.length()))
            .mapToObj(i -> new char[] { str1.charAt(i), str2.charAt(i) })
            .takeWhile(a -> a[0] < a[1])
            .map(a -> a[0])
            .collect(Collectors.toList());
    
    }
    

推荐阅读