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