匹配算法:向下就近原則,向下沒有就向上
- 實現方式一
- 實現方式二
- 總結
實現方式一
private static List<Integer> findMatches(List<Integer> sourceList, List<Integer> searchValues) {List<Integer> sortedList = sourceList.stream().filter(Objects::nonNull).sorted().collect(Collectors.toList());Set<Integer> foundValues = new HashSet<>();for (Integer searchValue : searchValues) {Integer nearestValue = findNearestBelowOrAbove(sortedList, searchValue);if (nearestValue != null) {foundValues.add(nearestValue);}}return sourceList.stream().filter(foundValues::contains).collect(Collectors.toList());}/*** @param sortedList* @param searchValue* @return 匹配結果(匹配規則:向下就近原則,向下沒有就向上)*/private static Integer findNearestBelowOrAbove(List<Integer> sortedList, Integer searchValue) {if (sortedList.isEmpty()) {return null;}int index = Collections.binarySearch(sortedList, searchValue);if (index >= 0) {return sortedList.get(index);} else {int insertionPoint = -(index + 1);if (insertionPoint < sortedList.size()) {return sortedList.get(insertionPoint);} else {return sortedList.get(sortedList.size() - 1);}}}
實現方式二
public static List<Integer> findMatches(List<Integer> sourceList, List<Integer> searchValues) {// 過濾 null 值并排序TreeSet<Integer> sortedSet = sourceList.stream().filter(Objects::nonNull).collect(Collectors.toCollection(TreeSet::new));Set<Integer> foundValues = new HashSet<>();for (Integer searchValue : searchValues) {Integer nearestValue = findNearestBelowOrAbove(sortedSet, searchValue);if (nearestValue != null) {foundValues.add(nearestValue);}}// 保持原始順序return sourceList.stream().filter(foundValues::contains).collect(Collectors.toList());}private static Integer findNearestBelowOrAbove(TreeSet<Integer> sortedSet, Integer searchValue) {// 查找大于等于 searchValue 的最小值Integer ceiling = sortedSet.ceiling(searchValue);if (ceiling != null) {return ceiling;}// 查找小于等于 searchValue 的最大值Integer floor = sortedSet.floor(searchValue);if (floor != null) {return floor;}return null;}
測試代碼:
public class TestMatcher {public static void main(String[] args) {List<Integer> sourceList = Arrays.asList(5, 7, 11, 31, 77);List<Integer> searchValues = Arrays.asList(1, 2, 8, 12, 13, 14, 15, 82, 91);List<Integer> matches = findMatches(sourceList, searchValues);System.out.println(matches);}
}
測試結果:
[5, 11, 31, 77]
總結
兩種實現的時間復雜度相同,都是 O(n log n)。
如果數據是靜態的,使用 基于排序列表的實現。
如果數據需要頻繁更新,使用 基于 TreeSet 的實現。