В .NET 4.0+ класс SortedSet<T>
имеет метод под названием GetViewBetween(l, r)
, который возвращает вид интерфейса на части дерева, содержащей все значения между указанными двумя. Учитывая, что SortedSet<T>
реализуется как красно-черное дерево, я, естественно, ожидаю, что он запустится в O(log N)
. Аналогичный метод в С++ - это std::set::lower_bound/upper_bound
, в Java он TreeSet.headSet/tailSet
, и они логарифмичны.
Однако это неверно. Следующий код запускается через 32 секунды, тогда как эквивалентная версия O(log N)
GetViewBetween
заставит этот код работать через 1-2 секунды.
var s = new SortedSet<int>();
int n = 100000;
var rand = new Random(1000000007);
int sum = 0;
for (int i = 0; i < n; ++i) {
s.Add(rand.Next());
if (rand.Next() % 2 == 0) {
int l = rand.Next(int.MaxValue / 2 - 10);
int r = l + rand.Next(int.MaxValue / 2 - 10);
var t = s.GetViewBetween(l, r);
sum += t.Min;
}
}
Console.WriteLine(sum);
Я декомпилировал System.dll, используя dotPeek, и вот что я получил:
public TreeSubSet(SortedSet<T> Underlying, T Min, T Max, bool lowerBoundActive, bool upperBoundActive)
: base(Underlying.Comparer)
{
this.underlying = Underlying;
this.min = Min;
this.max = Max;
this.lBoundActive = lowerBoundActive;
this.uBoundActive = upperBoundActive;
this.root = this.underlying.FindRange(this.min, this.max, this.lBoundActive, this.uBoundActive);
this.count = 0;
this.version = -1;
this.VersionCheckImpl();
}
internal SortedSet<T>.Node FindRange(T from, T to, bool lowerBoundActive, bool upperBoundActive)
{
SortedSet<T>.Node node = this.root;
while (node != null)
{
if (lowerBoundActive && this.comparer.Compare(from, node.Item) > 0)
{
node = node.Right;
}
else
{
if (!upperBoundActive || this.comparer.Compare(to, node.Item) >= 0)
return node;
node = node.Left;
}
}
return (SortedSet<T>.Node) null;
}
private void VersionCheckImpl()
{
if (this.version == this.underlying.version)
return;
this.root = this.underlying.FindRange(this.min, this.max, this.lBoundActive, this.uBoundActive);
this.version = this.underlying.version;
this.count = 0;
base.InOrderTreeWalk((TreeWalkPredicate<T>) (n =>
{
SortedSet<T>.TreeSubSet temp_31 = this;
int temp_34 = temp_31.count + 1;
temp_31.count = temp_34;
return true;
}));
}
Итак, FindRange
, очевидно, O(log N)
, но после этого мы вызываем VersionCheckImpl
... который выполняет линейный ход найденного поддерева только для пересчета его узлов!
- Зачем вам нужно все время проходить этот обход?
- Почему .NET не содержит метод
O(log N)
для разделения дерева на основе ключа, например С++ или Java? Это действительно полезно во многих ситуациях.