开启我的LeetCode刷题日记：341. 扁平化嵌套列表迭代器

Offer 驾到，掘友接招！我正在参与2022春招打卡活动，点击查看活动详情。

编程世界总是离不了算法

最近在看框架源码时，会有很多算法的实现逻辑，有时候会感到吃力

于是决定蹭着假期，加强算法和数据结构相关的知识

那怎么提升呢？

其实我知道算法这东西没有捷径，多写多练才能提升，于是我开启我的LeetCode刷题之旅

第一阶段目标是：200道，每天1到2篇

为了不乱，本系列文章目录分为三部分：

1. 今日题目：xxx
2. 我的思路
3. 代码实现

今天题目：341. 扁平化嵌套列表迭代器

给你一个嵌套的整数列表 nestedList 。每个元素要么是一个整数，要么是一个列表；该列表的元素也可能是整数或者是其他列表。请你实现一个迭代器将其扁平化，使之能够遍历这个列表中的所有整数。

实现扁平迭代器类 NestedIterator ：

NestedIterator(List nestedList) 用嵌套列表 nestedList 初始化迭代器。 int next() 返回嵌套列表的下一个整数。 boolean hasNext() 如果仍然存在待迭代的整数，返回 true ；否则，返回 false 。 你的代码将会用下述伪代码检测：

initialize iterator with nestedList res = [] while iterator.hasNext() append iterator.next() to the end of res return res 如果 res 与预期的扁平化列表匹配，那么你的代码将会被判为正确。

我的思路

• 数组的扁平化，正常来说就是先遍历一遍dfs，获取到完全扁平的数组\
• 然后hasnext判断数组长度\
• next输出执行slice()

代码实现

/**
 * @constructor
 * @param {NestedInteger[]} nestedList
 */
var NestedIterator = function(nestedList) {
    ans = []

    function dfs(list) {
      for(nest of list){
        if(nest.isInteger()){
          ans.push(nest.getInteger())
        }else {
          dfs(nest.getList())
        }
      }
    }
    dfs(nestedList)
};


/**
 * @this NestedIterator
 * @returns {boolean}
 */
NestedIterator.prototype.hasNext = function() {
    return ans.length > 0
};

/**
 * @this NestedIterator
 * @returns {integer}
 */
NestedIterator.prototype.next = function() {
    let val = ans[0]
    ans = ans.slice(1)
    return val
};

总结

实现方式其实有很多，这里仅供参考~

由于刚开始刷题，也不知道从哪里刷好，如果前辈们有好的建议，希望不吝赐教，感谢🌹