练习题 链表中的下一个更大节点

题目

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

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

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

示例 1：

输入：head = [2,1,5]
输出：[5,5,0]

示例 2：

输入：head = [2,7,4,3,5]
输出：[7,0,5,5,0]

提示：

• 链表中节点数为 n
• 1 <= n <= 104
• 1 <= Node.val <= 109

提交代码

//链表中的下一个更大节点  

//单调栈
//查找下一个更大节点
//下标从1开始
//可能没有下一个更大节点，没有则答案为0
//求的是下一个更大节点的值 
//单调栈里存储还未找到更大节点值的节点值 
//只能从左到右遍历 
//单调栈中存储的值越来越小 
//轮完所有节点后单调栈里还剩下的节点就是没有更大值的 

#include<iostream>
#include<vector>
#include<stack>
using namespace std;

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){}
};

struct Node{
	int num;//节点下标
	int val;//节点值 
};

ListNode* head;//链表 
vector<int> result(10000);//存储结果
stack<Node> stk;//单调栈 

//判断当前节点值是否是哪些节点的答案
void nln(int n,ListNode* node){
	while(!stk.empty()){
		if(node->val > stk.top().val){
			result[stk.top().num] = node->val;
			stk.pop();
		}else{
			break;
		}
	}
	
	stk.push(Node{n,node->val}); 
} 

int main(){
	//输入数组 
	/*int t;//存储整数 
	
	while(cin.peek() != '\n'){
		scanf("%d",&t);
		nums.push_back(t);
	}*/
	//输入链表
	head = new ListNode();
	ListNode* temp = head;
	int t;//存储节点值
	while(cin.peek() != '\n'){
		scanf("%d",&t);
		ListNode *next = new ListNode(t);
		temp->next = next;
		temp = next;
	}
	head = head->next;
	
	//-------------------------------
	
	//从左到右寻找下一个更大节点值
	temp = head;
	int count = 0;//第几个节点 
	while(temp){
		nln(count,temp);
		temp = temp->next;
		count++;
	} 
	//弹空单调栈，所有找不到的情况
	while(!stk.empty()){
		result[stk.top().num] = 0;
		stk.pop();
	}
	
	//输出结果
	for(int i = 0;i < count;i++){
		printf("%d ",result[i]);
	}
	
	return 0;
} 

 总结

解题思路：题目要求查找下一个更大节点的值，属于找下一个更大值的问题，可以使用单调栈解决。因为题目所给的数据结构为链表，所以只能从左向右应用单调栈，单调栈中存储的是尚未找到下一个更大值的节点值。从头到尾遍历链表，相当于每次尝试该节点值是否是左侧某个节点的答案，找到答案的弹出单调栈。链表遍历结束后，单调栈中剩余的节点对应的答案即为0（没有下一个更大值）。

注意：ListNode* node;表示声明了一个指针变量，指向ListNode类型的对象，这种声明表示node是一个指向ListNode对象的指针，通常在使用前需要为指针分配内存（如使用new）；ListNode node;表示声明了一个ListNode类型的对象，表示node是一个实际的ListNode对象，而不是指针。在使用这种声明时，对象会在栈上分配内存，而不需要使用new运算符显式分配内存。