LeetCode 1019

最新推荐文章于 2024-10-18 10:39:20 发布

小郭的学习日记

最新推荐文章于 2024-10-18 10:39:20 发布

阅读量43

点赞数

文章标签： leetcode 链表数据结构

本文链接：https://blog.csdn.net/weixin_46967754/article/details/130069069

版权

该文章描述了一种算法问题，即给定一个链表，对于每个节点找到其下一个更大的节点值。解决方案采用了单调栈，通过遍历链表，维护一个栈来保存节点，当遇到当前节点值大于栈顶节点值时，更新答案并出栈，最后返回答案数组。

摘要由CSDN通过智能技术生成

题目描述

给定一个长度为 n 的链表 head

对于列表中的每个节点，查找下一个更大节点的值。也就是说，对于每个节点，找到它旁边的第一个节点的值，这个节点的值严格大于它的值。

返回一个整数数组 answer ，其中 answer[i] 是第 i 个节点( 从1开始 )的下一个更大的节点的值。如果第 i 个节点没有下一个更大的节点，设置 answer[i] = 0 。

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/next-greater-node-in-linked-list
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路：

采用单调栈

代码：

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
    vector<int> nextLargerNodes(ListNode* head) {
        
        unordered_map<ListNode*,int> mp;

        ListNode* p = head;

        int index = 0 ;

        while(p)
        {
            mp[p] = index;
            index++;
            p=p->next;
        }
        
        vector<int> ans(index,0);
        stack<ListNode*> st;
        p=head;
        while(p)
        {
            
            while(!st.empty())
            {
                ListNode* cur = st.top();
                if(p->val > cur->val)
                {
                    index = mp[cur];
                    ans[index] = p->val;
                    st.pop();
                }
                else
                    break;
            }
            st.push(p);
            p=p->next;
        }
        return ans;
    }
};