From 22d81b8d8d9a068e60b0ef61e1f8c82beebfa9fb Mon Sep 17 00:00:00 2001 From: huangrui199126 Date: Sun, 28 Aug 2016 11:18:18 -0700 Subject: [PATCH] flattenArray --- Problems/src/usr/bin/flattenArray | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) create mode 100644 Problems/src/usr/bin/flattenArray diff --git a/Problems/src/usr/bin/flattenArray b/Problems/src/usr/bin/flattenArray new file mode 100644 index 0000000..a215aa5 --- /dev/null +++ b/Problems/src/usr/bin/flattenArray @@ -0,0 +1,16 @@ +#!/usr/bin/python +""" +var input = [1, {a: [2, [3]]}, 4, [5, [6]], [[7, ['hi']], 8, 9], 10]; +var output = [1, {a: [2, [3]]}, 4, 5, 6, 7, 'hi', 8, 9, 10]; +""" +def flattenArray(arr): + res = [] +# print arr + for item in arr: + if type(item) == type([]): + res += flattenArray(item) + else: + res.append(item) + return res + +print flattenArray([1, {"a": [2, [3]]}, 4, [5, [6]], [[7, ['hi']], 8, 9], 10])