次の範囲で、0 から 20 までの数値を検索します。
2-5 -> "a"
6-7 -> "b"
9-11 -> "c"
12-12 -> "d"
15-18 -> "e"
19-19 -> "f"
RangeCollection の作成と初期化を 1 回含む 1,000,000 回のループ内: わずか 3 秒!
あとは、Tuple の IEnumerable を準備するだけです。ここで、最初の int は最小 IP int 値、2 番目は最大値、TValue はその min~max 範囲に関連付けられたデータです。
これが私の実装です:
public class RangeCollection<TValue>
{
private readonly int[] _mins;
private readonly int[] _maxs;
private readonly TValue[] _values;
public RangeCollection(params Tuple<int, int, TValue>[] input)
: this((IEnumerable<Tuple<int, int, TValue>>)input)
{
}
public RangeCollection(IEnumerable<Tuple<int, int, TValue>> input)
{
var tuples = input.OrderBy(tuple => tuple.Item1).ToArray();
for (var i = 1; i < tuples.Length; i++)
{
if (tuples[i].Item1 <= tuples[i - 1].Item2)
{
throw new ArgumentException("Ranges should not overlap.");
}
}
this._mins = new int[tuples.Length];
this._maxs = new int[tuples.Length];
this._values = new TValue[tuples.Length];
for (var i = 0; i < tuples.Length; i++)
{
var tuple = tuples[i];
this._mins[i] = tuple.Item1;
this._maxs[i] = tuple.Item2;
this._values[i] = tuple.Item3;
}
}
public bool TryGetValue(int key, out TValue value)
{
if (this.Contains(key, out key))
{
value = this._values[key];
return true;
}
value = default(TValue);
return false;
}
public bool Contains(int key)
{
return this.Contains(key, out key);
}
private bool Contains(int key, out int index)
{
index = 0;
if (this._mins.Length == 0 || key < this._mins[0] || key > this._maxs[this._maxs.Length - 1])
{
return false;
}
var low = 0;
var high = this._mins.Length - 1;
while (high >= low)
{
index = (low + high) / 2;
var cmp = this._mins[index].CompareTo(key);
if (cmp == 0)
{
return true;
}
if (cmp == 1)
{
high = index - 1;
}
else
{
low = index + 1;
}
}
if (this._mins[index] > key)
{
index--;
}
else if (this._mins[index + 1] <= key)
{
index++;
}
return this._maxs[index] >= key;
}
}
使用法:
var collection = new RangeCollection<string>(new Tuple<int, int, string>(2, 5, "a"),
new Tuple<int, int, string>(6, 7, "b"),
new Tuple<int, int, string>(9, 11, "c"),
new Tuple<int, int, string>(12, 12, "d"),
new Tuple<int, int, string>(15, 18, "e"),
new Tuple<int, int, string>(19, 19, "f"));
for (var i = 0; i <= 20; i++)
{
string val;
if (collection.TryGetValue(i, out val))
{
Debug.WriteLine("Contains {0}. Value: {1}", i, val);
}
else
{
Debug.WriteLine("Doesn't contain " + i);
}
}
/* Output:
Doesn't contain 0
Doesn't contain 1
Contains 2. Value: a
Contains 3. Value: a
Contains 4. Value: a
Contains 5. Value: a
Contains 6. Value: b
Contains 7. Value: b
Doesn't contain 8
Contains 9. Value: c
Contains 10. Value: c
Contains 11. Value: c
Contains 12. Value: d
Doesn't contain 13
Doesn't contain 14
Contains 15. Value: e
Contains 16. Value: e
Contains 17. Value: e
Contains 18. Value: e
Contains 19. Value: f
Doesn't contain 20
*/