LeetCode解法汇总2487. 从链表中移除节点

目录链接:

力扣编程题-解法汇总_分享+记录-CSDN博客

GitHub同步刷题项目:

https://github.com/September26/java-algorithms

原题链接:

力扣(LeetCode)官网 - 全球极客挚爱的技术成长平台


描述:

给你一个链表的头节点 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, 105] 内
  • 1 <= Node.val <= 105

解题思路:

单调栈的解题思路。构造一个单调栈monotonicStack,这个栈存放单调非递增的节点。

遍历head中的节点,如果当前节点的值大于栈顶,则把栈顶移除,直至栈为空。然后把当前节点加入单调栈。

最后,利用单调栈反向生成链表,就是我们想要的结果。

代码:


class Solution {
public:
    ListNode *removeNodes(ListNode *head)
    {
        stack monotonicStack;

        ListNode *node = head;
        while (node != nullptr)
        {
            while (monotonicStack.size() > 0 && node->val > monotonicStack.top()->val)
            {
                monotonicStack.pop();
            }
            monotonicStack.push(node);
            node = node->next;
        }
        int i = 0;
        ListNode *header = nullptr;
        while (monotonicStack.size() > 0)
        {
            ListNode *node1 = monotonicStack.top();
            node1->next = header;
            header = node1;
            i++;
            monotonicStack.pop();
        }
        return header;
    }
};

你可能感兴趣的:(编程题,leetcode,算法,职场和发展)