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

給出一個以頭節點 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})

簡要題解

利用棧記錄一下下標來解決即可主要代碼如下:

/**
 * 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) {  
        vector<int>ans; stack<int>me;
        if(head==NULL)return ans;
       int i=0;
       while(head!=NULL)
       {
           while(!me.empty()&&head->val>ans[me.top()])
           {
               ans[me.top()]=head->val;
               me.pop();
           }
           ans.push_back(head->val);
           me.push(i++);
           head=head->next;
       }
       while(!me.empty())
       {
           ans[me.top()]=0;
           me.pop();
       }
       return ans;
    }
};
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章