leetcode 1019. 鏈表中的下一個更大節點

給出一個以頭節點 head 作爲第一個節點的鏈表。鏈表中的節點分別編號爲：node_1, node_2, node_3, ... 。

每個節點都可能有下一個更大值（next larger value）：對於 node_i，如果其 next_larger(node_i) 是 node_j.val，那麼就有 j > i 且  node_j.val > node_i.val，而 j 是可能的選項中最小的那個。如果不存在這樣的 j，那麼下一個更大值爲 0 。

返回整數答案數組 answer，其中 answer[i] = next_larger(node_{i+1}) 。

注意：在下面的示例中，諸如 [2,1,5] 這樣的輸入（不是輸出）是鏈表的序列化表示，其頭節點的值爲 2，第二個節點值爲 1，第三個節點值爲 5 。

 

示例 1：

輸入：[2,1,5]
輸出：[5,5,0]
示例 2：

輸入：[2,7,4,3,5]
輸出：[7,0,5,5,0]
示例 3：

輸入：[1,7,5,1,9,2,5,1]
輸出：[7,9,9,9,0,5,0,0]
 

提示：

對於鏈表中的每個節點，1 <= node.val <= 10^9
給定列表的長度在 [0, 10000] 範圍內

題解：單調隊列；

 

參考代碼：

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    vector<int> nextLargerNodes(ListNode* head) 
    {
        stack<pair<int,int>>s;
        map<int,int>m;
        int key=0;
        vector<int>ans;
        while(!s.empty())s.pop();        
        m.clear();  
        ans.clear();

        while(head)
        {
            while(!s.empty() && s.top().second<head->val)
            {
                m[s.top().first]=head->val;
                s.pop();
            }
            s.push(make_pair(key++,head->val));
            head=head->next;
        }
        for(int i=0;i<key;i++){
            ans.push_back(m[i]);
        }

        return ans;
    }
};