Автоматическое создание коллекции в словаре

Question

Много раз я должен создать Dictionary<KeyType, List<ValueType>>

Прежде чем я смогу начать использовать словарь, я должен сначала убедиться, что для этого ключа был создан список.

//Can i remove these two lines?
if(!dict.ContainsKey(key)) 
    dict[key]= new List<ValueType>;

//now use the key
dict[key].Add(value);

Я знаю только его "2 строки" кода, но это раздражает меня, и я думаю, что его можно удалить.

Я могу каким-то образом расширить словарь, но прежде чем я сделаю это, я хочу знать, нашел ли кто-то умный способ удалить приведенный выше оператор if.

В основном я хочу создать Dictionary<KeyType, Collection<ValueType>> и сразу начать использовать его, как dict[key].Add(value).

Answer 1

Вы можете создать что-то вроде Multimap ... Google Java Collection или просто добавить метод расширения, подобный этому:

public static void AddValue<TKey, TValue>
    (this IDictionary<TKey, List<TValue>> dictionary, TKey key, TValue value)
{
    List<TValue> values;
    if (!dictionary.TryGetValue(key, out values))
    {
        values = new List<TValue>();
        dictionary.Add(key, values);
    }
    values.Add(value);
}

Как говорит Беван, Lookup также может помочь, но вы можете создать его только с помощью метода ToLookup, и впоследствии вы не сможете его изменить. Во многих случаях это очень хорошая вещь, но если вам нужна изменяемая карта, вам понадобится что-то похожее на приведенное выше.

Answer 2

Метод ConcurrentDictionary<T,K>.GetOrAdd чрезвычайно полезен.

private ConcurrentDictionary<string, ICollection<int>> _dictionary;

private static ICollection<int> CreateEmptyList(string dummyKey)
{
    return new List<int>();
}

private void AddValue(string key, int value)
{
    ICollection<int> values = _dictionary.GetOrAdd(key, CreateEmptyList);
    values.Add(value);
}

Редактировать: Вот пример того, как реализовать функцию в качестве метода расширения для IDictionary<T,K> (C # 3):

Обратите внимание, что IDictionary<TKey, TValue>, как правило, не является потокобезопасным, поэтому, если вы хотите обеспечить безопасность потоков с помощью этого метода расширения, вам придется реализовать его вручную, как и для других операций.

public static TValue GetOrAdd<TKey, TValue>(
    this IDictionary<TKey, TValue> dictionary,
    TKey key,
    Func<TKey, TValue> valueFactory)
{
    TValue value;
    if (!dictionary.TryGetValue(key, out value))
    {
        value = valueFactory(key);
        dictionary.Add(key, value);
    }

    return value;
}

Answer 3

Взгляните на класс LookUp, представленный в Linq в .NET 3.5 - это может быть именно то, что вам нужно: класс Dictionary, который поддерживает несколько элементов на ключ.

Возможно, единственным существенным недостатком является то, что вам нужно, чтобы все ваши элементы были доступны в одном пакете, поскольку LookUp является неизменным.

Answer 4

Чтобы добавить к ответам, вы также можете добавить более общее расширение, которое принимает делегата для создания экземпляра:

public static TValue GetOrCreate<TKey, TValue>
    (this IDictionary<TKey, TValue> dict, 
          TKey key, 
          Func<TKey, TValue> getValue)
{
    TValue value;
    if (!dict.TryGetValue(key, out value))
    {
        dict.Add(key, getValue(key));
    }
    return value;
}

и тогда вы можете предоставить любой метод создания экземпляров, который вам нравится:

Dictionary<int, string> dict = new Dictionary<int, string>();
string result = dict.GetOrCreate(5, i => i.ToString());