Leetcode学习笔记（24. 两两交换链表中的节点）

最新推荐文章于 2024-10-18 20:06:06 发布

原创最新推荐文章于 2024-10-18 20:06:06 发布 · 288 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

LeetCode学习笔记专栏收录该内容

31 篇文章

订阅专栏

本文介绍了一种使用栈数据结构来实现链表中元素成对交换的算法。通过将每对节点压入栈中，再依次弹出以达到交换的目的，最后返回修改后的链表头指针。

在这里插入图片描述
用栈进行替换：

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */

class Solution {
public:
	ListNode* swapPairs(ListNode* head) {
		if (head == nullptr || head->next == nullptr)
			return head;
		ListNode* ptr = head;
		stack<ListNode*> data;
		ListNode* addhead = new ListNode(0);
		ListNode* p = addhead;
		int num = 0;
		while (ptr != nullptr) {
			if (num<2) {
				data.push(ptr);
				ptr = ptr->next;
				num++;
			}
			else {
				for (int i = 0; i<2; i++) {
					ListNode* temp = data.top();
					p->next = temp;
					p = temp;
					data.pop();
				}
				num = 0;
			}
		}
		while (!data.empty()) {
			ListNode* temp = data.top();
			p->next = temp;
			p = temp;
			data.pop();
		}
		p->next = nullptr;
		return addhead->next;
	}
};