lintcode 22.平面列表

来源:互联网 发布:长沙黑马程序员 编辑:程序博客网 时间:2024/06/08 15:27

  问你,何时曾看见,这世界为了人们改变!


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

 注意事项

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

样例

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

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

思路:大致就是做出一个判断,先判断给出的vector列表中的元素,如果是int类型,则放入新建立的vector中;

如果是列表性,则通过getline函数使其变为vector列表,不断调用,直到元素变为int类型,再放入刚开始建立的vector列表中。


  1. /**
     * // 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();
     * }
     */
    class Solution {  
    public:  
        // @param nestedList a list of NestedInteger  
        // @return a list of integer  
        vector<int> flatten(const vector<NestedInteger> &nestedList) {  
            vector<int> q;
            // Write your code here  
            int i;
            int n=nestedList.size();
            for(i=0;i<n;i++)
    {       
    if(nestedList[i].isInteger())
              q.push_back(nestedList[i].getInteger());
    else{
    vector<int> tmp=flatten(nestedList[i].getList());
    q.insert(q.end(),tmp.begin(),tmp.end());
    }
    }
    return q;
        }  
    };  
原创粉丝点击