Question

如标题所述，我们该怎么做？我觉得它应该类似于树遍历，但我无法弄清楚如何做到这一点。

    Map<Integer, Object> map = new HashMap<>();     

    map.put(1, 1);
    map.put(2, 2);


    map.put(3, new HashMap<Integer, Object>());
    Map<Integer, Object> m1 = (Map<Integer, Object>)map.get(3);
    m1.put(4, 4);
    m1.put(5, new HashMap<Integer, Object>());      
    m1.put(6, 6);

    Map<Integer, Object> m2 = (Map<Integer, Object>)m1.get(5);
    m2.put(8, 8);
    m2.put(9, 9);

    map.put(7, new HashMap<Integer, Object>());

在这个例子中，前两个＆＃34;值＆＃34;是两个整数，第三个＆＃34;值＆＃34;是一个嵌套的HashMap，它包含两个整数和一个更深层次的＆＃34;嵌套的HashMap。关于关键7，我想表达＆＃34;值＆＃34;也可以是一个空的HashMap - 因为这是一个面试问题，我想这可能会产生＆＃34;空值＆＃34;可能会影响next（）或hasNext（）实现？

Answer 1

实际上有很多不同的解决方案。到目前为止，这里最短的是使用Streams。

@SuppressWarnings("unchecked")
static <T> Stream<T> mapTraversalStream(Map<Integer, T> map) {
  return map.values().stream().flatMap(value -> {
    if(value instanceof Map) {
      return mapTraversalStream((Map<Integer, T>)value);
    } else {
      return Stream.of(value);
    }
  });
}

static <T> Iterator<T> mapTraversalIterator(Map<Integer, T> map) {
  return mapTraversalStream(map).iterator();
}

编辑：

如果您不能使用Java 8，您可以定义自己的地图并展平方法。

@SuppressWarnings("unchecked")
static <T> Iterator<T> mapTraversalIterator(Map<Integer, T> map) {
  return flatten(map(map.values().iterator(), new Function<T, Iterator<T>>() {
      public Iterator<T> apply(T value) {
        if(value instanceof Map) {
          return mapTraversalIterator((Map<Integer, T>)value);
        } else {
          return Collections.singleton(value).iterator();
        }
      }
  }));
}

interface Function<A, B> {
  B apply(A a);
}

static <A, B> Iterator<B> map(Iterator<A> iter, Function<A, B> op) {
  return new Iterator<B>() {
    public boolean hasNext() {
      return iter.hasNext();
    }
    public B next() {
      return op.apply(iter.next());
    }
  };
}

static <A> Iterator<A> flatten(Iterator<Iterator<A>> iter) {
  return new Iterator<A>() {
    private Iterator<A> next = null;

    public boolean hasNext() {
      while(iter.hasNext() && (next == null || !next.hasNext())) {
        next = iter.next();
      }
      if(next == null || !next.hasNext()) return false;
      else return true;
    }
    public A next() {
      if(!hasNext()) throw new NoSuchElementException();
      return next.next();
    }
  };
}

为嵌套的HashMap实现Java Iterator，其中元素可以是1）整数或2）另一个嵌套的HashMap，或者3）空的HashMap

1 个答案: