迭代时间复杂度为O（log（N））的TreeSet

Question

我目前有一个WorkShift的TreeSet。 WorkShift实例包含以下信息和方法：

package work;


public class WorkShift {

    private final int start;

    private final int end;

    private final int duration;

    /**
     * Constructor.
     * 
     * @param start 
     * @param end 
     */
    public TTInterval(int start, int end) {
        this.start = start;
        this.end = end;
        this.duration = end - start + 1; //
    }

    /**
     * @return Start
     */
    public int getStart() {
        return start;
    }

    /**
     * @return End
     */
    public int getEnd() {
        return end;
    }

    /**
     * @return Duration
     */
    public int getDuration() {
        return duration;
    }
}

一个WorkDay由WorkShift 。 这是一个名为shifts of WorkShifts的TreeSet，它在start时间之后排序。 它的类文件如下所示：

package work;

import java.util.Arrays;
import java.util.Comparator;
import java.util.TreeSet;

public class WorkDay {
    /**
     * Set of shifts sorted by their start time.
     */
    private final TreeSet<WorkShift> shifts =
            new TreeSet<>(Comparator.comparing(WorkShift::getStart));

}

现在，我想添加一个方法getFirstShiftByDuration ，该方法获取一个int duration作为参数并返回具有该持续时间的shifts集的第一个班次，同时保持最大时间复杂度O（log（N）） 。

“空” getFirstShiftByDuration如下所示：

    /**
     * Returns the workShift with the duration we are looking for
     * 
     * @param duration
     * @pre duration > 0
     * 
     * @return First workShift found with that duration.
     */
    public workShift getfirstWorkShiftByDuration(int duration) {
        assert duration > 0;

        /*TODO*/
    }

通常，我会创建一个for循环，遍历该集合并查找该特定持续时间。 但是，由于现在存在O（log（N））的运行时限制，因此我在选择和可以使用的预实现方法上受到很大限制。

该集合本身还由仅由其start和end定义的元素组成。 我无法弄清楚如何使用预定义的TreeSet方法，甚至在它不是属性时也可以通过它们的持续时间进行比较。

有什么方法可以解决此问题，而无需创建新类并超出O（log（N））运行时限制？