22. Flatten List

最新推荐文章于 2022-07-01 20:42:46 发布

jingxindeyi

最新推荐文章于 2022-07-01 20:42:46 发布

阅读量434

点赞数

CC 4.0 BY-SA版权

分类专栏： # 刷题

本文链接：https://blog.youkuaiyun.com/zslngu/article/details/88432193

刷题专栏收录该内容

105 篇文章

订阅专栏

本文介绍了一种将嵌套列表转换为单一整数列表的算法，通过迭代而非递归方式实现，有效解决多层次嵌套问题。

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

22. Flatten List
中文English
Given a list, each element in the list can be a list or integer. flatten it into a simply list with integers.

Example
Example 1:
	Input:  [[1,1],2,[1,1]]
	Output: [1,1,2,1,1]
	
	Explanation:
	flatten it into a simply list with integers.

Example 2:
	Input:  [1,2,[1,2]]
	Output:[1,2,1,2]
	
	Explanation:  
	flatten it into a simply list with integers.

Example 3:
	Input: [4,[3,[2,[1]]]]
	Output:[4,3,2,1]
	
	Explanation: 
	flatten it into a simply list with integers.

Challenge
Do it in non-recursive.

Notice
If the element in the given list is a list, it can contain list too.

class Solution(ob
ject):

    # @param nestedList a list, each element in the list 
    # can be a list or integer, for example [1,2,[1,2]]
    # @return {int[]} a list of integer
    def flatten(self, nestedList):
        # Write your code here
        if type(nestedList) == int:
            return [nestedList]
        s = []
        result = []
        it = iter(nestedList)
        while True:
            try:
                t = next(it)
                if type(t) == int:
                    result.append(t)
                else:
                    s.append(it)
                    it = iter(t)
            except StopIteration:
                if len(s) == 0:
                    break
                it = s.pop()
        return result