根据另一个`List`过滤`List` [重复]

Question

我正在使用下面的函数来过滤基于另一个列表的列表。例如：programLines [1, 2, 4, 5, 5, 6, 6, 7, 7, 9, 13] 和行 [9, 10, 13, 14] 到此输出 [9, 13]。但是我想将其更改为此输出--> [9,0,13,0] 换句话说，以填充值为0 的其余元素。

public List<Integer> filter(List<Integer> lines, List<Integer> programLines)              
            List<Integer> listOutput =
                lines.stream().filter(programLines::contains).collect(Collectors.toList());
            return listOutput;
          }

Answer 1

使用map 代替filter：

public List<Integer> filter(List<Integer> lines, List<Integer> programLines)              
    List<Integer> listOutput =
        lines.stream()
             .map(i -> programLines.contains(i) ? i : 0)
             .collect(Collectors.toList());
    return listOutput;
}

如果programLines 很大，将其转换为Set 可能会有所帮助以获得更好的性能：

public List<Integer> filter(List<Integer> lines, List<Integer> programLines)              
    Set<Integer> filter = new HashSet<>(programLines);
    List<Integer> listOutput =
        lines.stream()
             .map(i -> filter.contains(i) ? i : 0)
             .collect(Collectors.toList());
    return listOutput;
}