22. 平面列表-LintCode

22.平面列表

题目

给定一个列表，该列表中的每个要素要么是个列表，要么是整数。将其变成一个只包含整数的简单列表。

 注意事项

如果给定的列表中的要素本身也是一个列表，那么它也可以包含列表。

您在真实的面试中是否遇到过这个题？ 

Yes

  样例

   给定 [1,2,[1,2]]，返回 [1,2,1,2]。

   给定 [4,[3,[2,[1]]]]，返回 [4,3,2,1]。

   解题思路

   使用递归操作, 并使用vector容器类对象的方法去获取其中的内层数字.

   参考代码

class Solution {
public:
    // @param nestedList a list of NestedInteger
    // @return a list of integer
    vector<int> vec;
    vector<int> flatten(const vector<NestedInteger> &nestedList) 
    {
        // Write your code here
        int size = nestedList.size();
        for (int i = 0; i < size; i++) 
        {
            if (nestedList[i].isInteger()) 
            {
                vec.push_back(nestedList[i].getInteger());
            } 
            
            else 
            {
               flatten(nestedList[i].getList());
            }
        }
        return vec;
    }
};