341. 扁平化嵌套列表迭代器_easyui 扁平化组件-优快云博客

本文链接：https://blog.youkuaiyun.com/csdnyq/article/details/100804509

本文介绍了一种深度优先遍历策略，用于解决嵌套整型列表的扁平化问题。通过设计迭代器，可以遍历整型列表中的所有整数，无论其是否嵌套在子列表中。文章提供了具体的实现代码，展示了如何处理嵌套结构，并确保正确输出顺序。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

341. 扁平化嵌套列表迭代器

题目描述

给定一个嵌套的整型列表。设计一个迭代器，使其能够遍历这个整型列表中的所有整数。

列表中的项或者为一个整数，或者是另一个列表。

示例 1:

输入: [[1,1],2,[1,1]]
输出: [1,1,2,1,1]
解释: 通过重复调用 next 直到 hasNext 返回false，next 返回的元素的顺序应该是: [1,1,2,1,1]。

示例 2:

输入: [1,[4,[6]]]
输出: [1,4,6]
解释: 通过重复调用 next 直到 hasNext 返回false，next 返回的元素的顺序应该是: [1,4,6]。

思路

可以把嵌套数组看成一棵树，使用深度优先遍历。

实现

/**
 * // This is the interface that allows for creating nested lists.
 * // You should not implement it, or speculate about its implementation
 * public interface NestedInteger {
 *
 *     // @return true if this NestedInteger holds a single integer, rather than a nested list.
 *     public boolean isInteger();
 *
 *     // @return the single integer that this NestedInteger holds, if it holds a single integer
 *     // Return null if this NestedInteger holds a nested list
 *     public Integer getInteger();
 *
 *     // @return the nested list that this NestedInteger holds, if it holds a nested list
 *     // Return null if this NestedInteger holds a single integer
 *     public List<NestedInteger> getList();
 * }
 */
public class NestedIterator implements Iterator<Integer> {
    private List<Integer> seq = new ArrayList<>();
    private int count = 0;

    public NestedIterator(List<NestedInteger> nestedList) {
        dfs(nestedList);
    }
    
    /**
    * 深度优先遍历
    */
    public void dfs(List<NestedInteger> nestedList) {
        for(NestedInteger node : nestedList) {
            if(node.isInteger()) {
                seq.add(node.getInteger());
            } else {
                dfs(node.getList());
            }
        }
    }

    @Override
    public Integer next() {
        return seq.get(count++);
    }

    @Override
    public boolean hasNext() {
        return count < seq.size();
    }
}

/**
 * Your NestedIterator object will be instantiated and called as such:
 * NestedIterator i = new NestedIterator(nestedList);
 * while (i.hasNext()) v[f()] = i.next();
 */