Careercup

2014-05-12 06:19

题目链接

原题：

Given an input list of lists.. flatten the list. For e.g. 

{{1,2}, {3}, {4,5}} ... Output should be {1, 2, 3, 4, 5}

题目：给定一个二维数组，把它展开成一维数组。

解法：之前做了一道题，把维度不定的数组展开成一维数组。这题也能用那题的代码来解决。

代码：

# http://www.careercup.com/question?id=5485521224597504
#!/usr/bin/python

def flatten(container, new_container):
    for element in container:
        if isinstance(element, list) or isinstance(element, tuple) or isinstance(element, set):
            flatten(element, new_container)
        else:
            new_container.append(element)
    pass
    
if __name__ == '__main__':
    a = [1, 2.3, [3, [1, 222]], (4, 111, 0), "string", set([1, "hello", 3.33])]
    res = []
    flatten(a, res)
    print(res)
    pass