基於Python和C++實現刪除連結串列的節點
阿新 • • 發佈:2020-07-07
給定單向連結串列的頭指標和一個要刪除的節點的值,定義一個函式刪除該節點。
返回刪除後的連結串列的頭節點。
示例 1:
輸入: head = [4,5,1,9],val = 5
輸出: [4,9]
解釋: 給定你連結串列中值為 5 的第二個節點,那麼在呼叫了你的函式之後,該連結串列應變為 4 -> 1 -> 9.
示例 2:
輸入: head = [4,val = 1
輸出: [4,9]
解釋: 給定你連結串列中值為 1 的第三個節點,那麼在呼叫了你的函式之後,該連結串列應變為 4 -> 5 -> 9.
思路:
建立一個空節點作為哨兵節點,可以把首尾等特殊情況一般化,且方便返回結果,使用雙指標將更加方便操作連結串列。
Python解法:
class ListNode: def __init__(self,x): self.val = x self.next = None class Solution: def deleteNode(self,head: ListNode,val: int) -> ListNode: tempHead = ListNode(None) # 構建哨兵節點 tempHead.next = head prePtr = tempHead # 使用雙指標 postPtr = head while postPtr: if postPtr.val == val: prePtr.next = postPtr.next break prePtr = prePtr.next postPtr = postPtr.next return tempHead.next
C++解法:
struct ListNode { int val; ListNode *next; ListNode(int x) : val(x),next(NULL) {} }; class Solution { public: ListNode* deleteNode(ListNode* head,int val) { ListNode *tempHead = new ListNode(-1); // 哨兵節點,建立節點一定要用new!!!!!!!!!!!!!! tempHead->next = head; ListNode *prePtr = tempHead; ListNode *postPtr = head; while (postPtr) { if (postPtr->val == val) { prePtr->next = postPtr->next; // 畫圖確定指標指向關係,按照箭頭確定指向 break; } postPtr = postPtr->next; prePtr = prePtr->next; } return tempHead->next; } };
以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支援我們。