我正在编写一个特定的优先级队列。其结构需要如下所示:
Priority(<int>) Data(List<Object>)
1 a, b, g, h
3 c, d, j
4 k
10 e, f, i
我需要能够有效地查找给定优先级是否存在列表;如果没有,请创建列表并添加消息,否则将消息附加到现有列表中。
我写了一棵红黑树,但这似乎有点矫枉过正,可能不是最快的解决方案。它还有一个缺点,即无法按优先级轻松抓取消息,我需要在编写完成后能够执行此操作。
我想到了字典,但除非我弄错了,否则它没有一个简单的方法来说"如果键__存在,给我它对应的值,否则给我null"。还是我错过了什么?
编辑
我目前的实现是有 32 个固定列表。适用的列表将添加到 32 位标志中,并在 32 位标志中设置适用的位。我使用De Bruijn的算法来获取LSB。这很有效,但增加了我想要减轻的其他复杂性。
也许你应该使用Dictionary<int,List<object>>
public void Add(int priority,object data)
{
if(dictionary.ContainsKey(priority))
dictionary[priority].Add(data);
else
dictionary.Add(priority,new List<object>{data});
}
SortedDictionary 可以填补这项工作。只需使用 TryGetValue() 有条件地查找列表。
嗯,Dictionary
作为底层容器有什么问题?平均而言,您可以获得 O(1) 访问/插入时间,而不是使用 rb 树获得 O(log n)。只需根据您的需要包装Dictionary
,例如:
internal public class PriorityQueue<TValue>
{
private Dictionary<int, List<TValue>> mDict;
// only Add, TryGetValue shown...
public void Add(int pPriority, TValue pInput)
{
List<TValue> tTmp;
if (mDict.TryGetValue(pPriority, tTmp))
{
tTmp.Add(pInput);
}
else
{
mDict.Add(pPriority, new List<TValue>{ pInput });
}
}
public bool TryGetValue(int pPriority, out List<TValue>)
{
// obvious...
}
}