Многоуровневая карта в Java

Каков наилучший способ в Java сохранить значения ("o") в древовидной структуре следующим образом:

                    obj1                 
                     /\
                    /  \
                   /    \
              obj2        obj3
              /\            /\
             /  \          /  \
            /    \        /    \
          obj4  obj5    obj6   obj7
          /\     /\     /\      /\
         /  \   /  \   /  \    /  \
        o8   oN...

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

Мне нужно получить какой-то список значений ключами - точно так же, как на моем снимке. Другими словами, структура не должна становиться плоской.

Мне нужно .get(obj3) чтобы вернуть {obj6, obj7},.get(obj1) - {obj2, obj3}.

На данный момент я использую Map для этого, но раздувание такой карты является уродливым, потому что мне нужно проверить каждый уровень структуры. Похоже, что (данные - это карта):

if(data.get(somedouble) == null) {
    Map<Integer, Data> inm = new TreeMap<>();
    inm.put(someint, obj);
    Map<Double, Map<Integer, Data>> m = new TreeMap<>();
    m.put(somedouble2, inm);
    data.put(somedouble, m);
}
else {
    if(data.get(somedouble).get(somedouble2) == null) {
        Map<Integer, Data> inm = new TreeMap<>();
        inm.put(someint, obj);
        data.get(somedouble).put(somedouble2, inm);
    }
    else
        data.get(somedouble).get(somedouble2).put(someint, obj);
}

Производительность не проблема, но красота кода.

Ответ 1

Вы можете использовать свой конкретный ключ:

class MyKey {
    Double beta;
    Double yaw;
    int minute;

    public int hashCode() {
        /* Returns hash code from a combination of hash of the key members. */
    }

    @Override
    public boolean equals(Object obj) {
        /* Returns true if obj is a MyKey with same members. */
    }
}

А потом просто:

data.put(myKey, obj);

Таким образом, "многоуровневые проверки" скрыты в MyKey.equals(). Он держит код клиента чистым, а сложность ключа находится в надежном месте.

Изменить изменения после изменения:

Если, помимо всего прочего, вы хотите иметь карту из своей двойной beta на свои объекты, тогда я все равно буду придерживаться плана такого плана.

Фактически вы хотите иметь несколько "индексов" для ваших данных, например, в базе данных, чтобы вы могли запрашивать объекты с такими же "бета" или "рысканием". Для этого лучше всего использовать несколько Map (фактически Multimap), по одному для каждого из ваших "индексов".

Использование мультикапа Guava:

ListMultimap<Double, Data> mapForBeta;
ListMultimap<Double, Data> mapForYaw;

Вы можете поместить все мультимары и Map<MyKey, Data> в ваш конкретный класс. На самом деле лучшим способом было бы подкласс Map<MyKey, Data>:

public class MyMap extends HashMap<MyKey, Data> {

    ListMultimap<Double, Data> mapForBeta;
    ListMultimap<Double, Data> mapForYaw;


    public Data put(MyKey key, Data value) {
        super.put(key, value);
        mapForBeta.add(key.beta, value);
        mapForYaw.add(key.yaw, value);
    };

    public List<Data> getFromBeta(Double beta) {
        return mapForBeta.get(beta);
    }

    public List<Data> getFromYaw(Double yaw) {
        return mapForYaw.get(yaw);
    }
}

Новое редактирование с лучшим решением:

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

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

public class DefaultMap<K, V> extends TreeMap<K, V> {

    static abstract class Generator<V>{
        abstract V create();
    }

    final Generator<V> generator;


    DefaultMap(Generator<V> generator) {
        this.generator = generator;
    }

    @Override
    public V get(Object key) {
        V val = super.get(key);
        if (val == null) {
            val = generator.create();

            put((K)key, val);
        }

        return val;
    }
}

Теперь вы можете иметь свой древовидный класс утилиты для хранения всех ваших данных:

public class MyTree {
  private final Map<Double, Map<Double, Map<Integer, Data>>> data;

  public MyTree() {
    data = new DefaultMap<>(new Generator<Map<Double, Map<Integer, Data>>>() {
      @Override
      Map<Double, Map<Integer, Data>> create() {
        return new DefaultMap<>(new Generator<Map<Integer, Data>>() {

          @Override
          Map<Integer, Data> create() {
            return new TreeMap<>();
          }

        });
      }
    });
  }

  void add(MyKey d, Data obj) {
    data.get(d.beta).get(d.yaw).put(d.minute, obj);
  }
}

Теперь вы можете получить доступ к своим данным с помощью data.get(beta).get(yaw), и у вас нет кода спагетти для хранения ваших значений.

Ответ 2

Как насчет двоичного дерева?

Node.java

public class Node {

    private Node leftChild;
    private Node rightChild;

    public Node(Node leftChild, Node rightChild) {
        this.leftChild = leftChild;
        this.rightChild = rightChild;
    }

    public boolean isLeaf() {
        return this instanceof Leaf;
    }

}

Leaf.java

public class Leaf extends Node {

    private Data data;

    public Leaf(Data data) {
        super(null, null);
        this.data = data;
    }

}

Например, если вы хотите построить следующее дерево:

         root
          /\
         /  \
        /    \
   node1     node2
   /\           /\
  /  \         /  \
leaf1 leaf2  leaf3 leaf4

Использование:

Leaf leaf1 = new Leaf(new Data());
Leaf leaf2 = new Leaf(new Data());
Leaf leaf3 = new Leaf(new Data());
Leaf leaf4 = new Leaf(new Data());

Node node1 = new Node(leaf1, leaf2);
Node node2 = new Node(leaf3, leaf4);

Node root = new Node(node1, node2);