大佬们，假设下面这个是已经排序过的关于时间的数组，现在需要添加到不同距离当前年份时间的数组，能用二分查找去实现吗？

[unregister]

public void test(){ String time2 = "2022-08-31"; for(int i = 0; i<times.size();i++){ String time = times.get(i); if(betweenMonth(time,time2) < 12){ oneYearTimes.add(time); } else if(betweenMonth(time,time2) >= 12 && betweenMonth(time,time2) < 24 ){ one2twoYearTimes.add(time); } else if(betweenMonth(time,time2) >= 24 && betweenMonth(time,time2) < 36 ){ two2threeYearTimes.add(time); } else if(betweenMonth(time,time2) >= 36 && betweenMonth(time,time2) < 48 ){ three2fourYearTimes.add(time); } else if(betweenMonth(time,time2) >= 48 && betweenMonth(time,time2) < 48 ){ four2fiveYearTimes.add(time); } else if(betweenMonth(time,time2) >= 60 ){ moreThanFiveYearTimes.add(time); } } }

static Long betweenMonth(String time1, String time2){
    return ChronoUnit.MONTHS.between(parse(time1), parse(time2));
}

static LocalDate parse(String str){
    DateTimeFormatter df = DateTimeFormatter.ofPattern("yyyy-MM-dd");
    return LocalDate.parse(str, df);
}

[JasonLaw]

@unregister #18 我不是大佬，我还是在一直学习。如果你想要更加理解代码的话，真的建议你学习一下算法和数据结构，https://neetcode.io/是个好地方，我也在使用这个网站，大家一起进步。

[aguesuka]

首先你的代码应该是这样, 也就是先把重复的代码抽离成方法.

public void test() {
String time2 = "2022-08-31";
for (int i = 0; i < times.size(); i++) {
String time = times.get(i);
var monthsBetween = betweenMonth(time, time2);
findTimeList(monthsBetween).add(time);
}
}

List<String> findTimeList(long monthsBetween){
// TODO
}

[zmal]

@JasonLaw
private int 返回入参时间距今几年(String time);
Map<Integer, List<String>> yearMap = list.stream().collect(Collectors.groupingBy(this::返回入参时间距今几年, TreeMap::new, Collectors.toList()));

treeMap 只是用来把年份做个排序，遍历的时候方便。不用也可以。

[msg7086]

你想把扫全表变成二分搜来做 grouping 是吗？

确实可以做的。

比如说 Ruby 里有 bsearch_index 可以通过条件在数组里做二分搜，返回第一个满足条件的项的下标。
假设 times 是时间数组，降序，time2 是比较日期
year2_date = time2 - 1.year
year2_idx = times.bsearch_index {|time| time < year2_date} #找到第一个 1 年以前的日期
year3_date = time2 - 2.year
year3_idx = times.bsearch_index {|time| time < year3_date} #找到第一个 2 年以前的日期
然后把 0-year2_idx 的和 year2_idx-year3_idx 等等的 slice 切出来放进不同数组就行了。

[JasonLaw]

@zmal #23 有一个疑问，使用 tree map 的话，时间复杂度是 O(nlgn)，对吗？

[zmal]

@JasonLaw
？？？ treeMap 的 key 是 int 值的年份，总共才 5 个数，和时间集合的长度无关。

[JasonLaw]

@zmal #26 对，如果不包含所有年份差的话，就是只有 6 个 key 。

[unregister]

@JasonLaw Thank you,已经收藏啦
@aguesuka 学习了。
@msg7086 谢谢大佬的思路。