LeetCode-24:Swap Nodes in Pairs(兩兩交換連結串列中的節點) -- Medium
阿新 • • 發佈:2018-12-31
題目:
Given a linked list, swap every two adjacent nodes and return its head.
例子:
Example 1:
Given 1->2->3->4, you should return the list as 2->1->4->3.
Note:
- Your algorithm should use only constant extra space.
- You may not modify the values in the list’s nodes, only nodes itself may be changed.
問題解析:
給定一個連結串列,兩兩交換其中相鄰的節點,並返回交換後的連結串列。
相似題:
- Reverse Nodes in k-Group
思路標籤
棧的思想
解答:
棧
- 和第25題是相似的題目,因為連結串列是以k長度,這裡(k=2)進行翻轉,所以最後生成的連結串列是k大小的先進後出,則明顯可以以棧的形式來實現。
- 注意要加一個返回連結串列頭節點的指標的指標。
- 同時需要處理末尾最後小於k的節點。
/**
* 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) {
stack<ListNode*> k;
ListNode *cur = head, *lead = new ListNode(0), *prv =lead;
while(cur){
if(k.size()==2){
while(k.size()){
prv->next = k.top();
k.pop();
prv = prv->next;
prv->next = NULL;
}
}
k.push(cur);
cur = cur->next;
}
while (k.size()){
prv->next = k.top();
k.pop();
prv = prv->next;
prv->next = NULL;
}
return lead->next;
}
};