列表扁平化

描述

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

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

样例

样例 1:

输入:

列表 = [[1,1],2,[1,1]]

输出:

[1,1,2,1,1]

解释:将其变成一个只包含整数的简单列表。

样例 2:

输入:

列表 = [1,2,[1,2]]

输出:

[1,2,1,2]

解释:将其变成一个只包含整数的简单列表。

样例 3:

输入:

列表 = [4,[3,[2,[1]]]]

输出:

[4,3,2,1]

解释:将其变成一个只包含整数的简单列表。

比较直观采用DFS做,直接上代码。

/**
 * // 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();
 * }
 */
public class Solution {
public List<Integer> flatten(List<NestedInteger> nestedList) {
            List<Integer> list = new ArrayList<>();
            helper(nestedList,list);
            return list;
        }
        public void helper(List<NestedInteger> nestedList,List<Integer> list){
            if(nestedList.size()==0){
                return;
            }
            for (NestedInteger nestedInteger:nestedList) {
                if (nestedInteger.isInteger()){
                    list.add(nestedInteger.getInteger());
                }else{
                    helper(nestedInteger.getList(),list);
                }
            }
        }
      }

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值