[OC]题解：链表中的下一个更大节点

最新推荐文章于 2024-10-17 16:33:41 发布

OC溥哥999

最新推荐文章于 2024-10-17 16:33:41 发布

阅读量60

点赞数

分类专栏：一本通提高文章标签：链表数据结构

本文链接：https://blog.csdn.net/m0_73923553/article/details/130045610

版权

一本通提高专栏收录该内容

5 篇文章 0 订阅

订阅专栏

该程序实现了一个C++解决方案，用于解决链表问题，即给定一个链表，找到每个节点的下一个更大值。它使用一个栈来存储已遍历的节点值，然后从后往前遍历链表，如果栈顶的值小于当前节点值，则更新答案并弹出栈顶元素。最后，返回包含每个节点的下一个更大值的数组。

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

题目描述

给出一个以头节点 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 。

样例输入 Copy

【样例输入1】
3
2 1 5
【样例输入2】
5
2 7 4 3 5

样例输出 Copy

【样例输出1】
5 5 0
【样例输出2】
7 0 5 5 0

提示

对于链表中的每个节点，1 <= node.val <= 10^9
给定列表的长度在 [0, 10000] 范围内

题解：

#include<bits/stdc++.h>
using namespace std;
struct ListNode {
	int val;
	ListNode* next;
	ListNode() : val(0), next(NULL) {}
	ListNode(int x) : val(x), next(NULL) {}
	ListNode(int x, ListNode* next) : val(x), next(next) {}
};
class Solution {
	public:
		vector<int> nextLargerNodes(ListNode* head) {
			vector<int> res;
			ListNode* p=head;
			while(p) {
				res.push_back(p->val);
				p=p->next;
			}
			int n=res.size();
			stack<int> tmp;
			vector<int> ans(n);
			for(int i=n-1; i>=0; i--) {
				while(!tmp.empty() && tmp.top()<=res[i]) {
					tmp.pop();
				}
				if(!tmp.empty()) {
					ans[i]=tmp.top();
				} else {
					ans[i]=0;
				}
				tmp.push(res[i]);
			}
			return ans;
		}
};
ListNode* createByTail() {
	ListNode* head;
	ListNode* p1,*p2;
	int n=0,num;
	int len;
	cin>>len;
	head=NULL;
	while(n<len && cin>>num) {
		p1=new ListNode(num);
		n=n+1;
		if(n==1)head=p1;
		else p2->next=p1;
		p2=p1;
	}
	return head;
}
int main() {
	ListNode* head = createByTail();
	vector<int> res=Solution().nextLargerNodes(head);
	for(int i=0; i<res.size(); i++) {
		if (i>0) cout<<" ";
		cout<<res[i];
	}
	cout<<endl;
	return 0;
}

三联，谢谢