LeetCode 2487. 从链表中移除节点:单调栈

【LetMeFly】2487.从链表中移除节点:单调栈

力扣题目链接:https://leetcode.cn/problems/remove-nodes-from-linked-list/

给你一个链表的头节点 head

移除每个右侧有一个更大数值的节点。

返回修改后链表的头节点head

示例 1:

复制代码
输入:head = [5,2,13,3,8]
输出:[13,8]
解释:需要移除的节点是 5 ,2 和 3 。
- 节点 13 在节点 5 右侧。
- 节点 13 在节点 2 右侧。
- 节点 8 在节点 3 右侧。

示例 2:

复制代码
输入:head = [1,1,1,1]
输出:[1,1,1,1]
解释:每个节点的值都是 1 ,所以没有需要移除的节点。

提示:

  • 给定列表中的节点数目在范围 [1, 10^5^]
  • 1 <= Node.val <= 10^5^

方法一:单调栈

维护一个单调递减栈(严格地说是单调非递增栈):

遍历链表,在当前节点大于栈顶节点时不断弹出栈顶节点,然后将当前节点入栈。

最终,从栈底到栈顶的元素就是非递增的了。因此也就得到了想要的链表。

  • 时间复杂度 O ( l e n ( l i s t n o d e ) ) O(len(listnode)) O(len(listnode))
  • 空间复杂度 O ( l e n ( l i s t n o d e ) ) O(len(listnode)) O(len(listnode))

然后被丢弃节点的delete操作就靠力扣了hh。

AC代码

C++
cpp 复制代码
class Solution {
public:
    ListNode* removeNodes(ListNode* head) {
        stack<ListNode*> st;
        while (head) {
            while (st.size() && st.top()->val < head->val) {
                st.pop();
            }
            st.push(head);
            head = head->next;
        }
        ListNode* lastNode = nullptr;
        while (st.size()) {
            ListNode* thisNode = st.top();
            st.pop();
            thisNode->next = lastNode;
            lastNode = thisNode;
        }
        return lastNode;
    }
};
Python
python 复制代码
class Solution:
    def removeNodes(self, head: ListNode) -> ListNode:
        st = []
        while head:
            while len(st) and st[-1].val < head.val:
                st.pop()
            st.append(head)
            head = head.next
        for i in range(len(st) - 1):
            st[i].next = st[i + 1]
        return st[0]

同步发文于CSDN,原创不易,转载经作者同意后请附上原文链接哦~

Tisfy:https://letmefly.blog.csdn.net/article/details/135357617

相关推荐
小码农<^_^>8 分钟前
优选算法精品课--滑动窗口算法(一)
算法
羊小猪~~10 分钟前
神经网络基础--什么是正向传播??什么是方向传播??
人工智能·pytorch·python·深度学习·神经网络·算法·机器学习
软工菜鸡35 分钟前
预训练语言模型BERT——PaddleNLP中的预训练模型
大数据·人工智能·深度学习·算法·语言模型·自然语言处理·bert
南宫生38 分钟前
贪心算法习题其三【力扣】【算法学习day.20】
java·数据结构·学习·算法·leetcode·贪心算法
AI视觉网奇1 小时前
sklearn 安装使用笔记
人工智能·算法·sklearn
JingHongB1 小时前
代码随想录算法训练营Day55 | 图论理论基础、深度优先搜索理论基础、卡玛网 98.所有可达路径、797. 所有可能的路径、广度优先搜索理论基础
算法·深度优先·图论
weixin_432702261 小时前
代码随想录算法训练营第五十五天|图论理论基础
数据结构·python·算法·深度优先·图论
小冉在学习1 小时前
day52 图论章节刷题Part04(110.字符串接龙、105.有向图的完全可达性、106.岛屿的周长 )
算法·深度优先·图论
Repeat7151 小时前
图论基础--孤岛系列
算法·深度优先·广度优先·图论基础
小冉在学习2 小时前
day53 图论章节刷题Part05(并查集理论基础、寻找存在的路径)
java·算法·图论