NOTE01:要考慮的是,因為第一個節點也有可能被刪掉 -> 要做虛擬節點
NOTE02:他給的是【倒數要刪除的點】,那我們要先知道整理長度,才可以去算出【順向要刪除的是第幾個點】
class Solution {
public:
int getLength(ListNode* head){
int ans = 0;
while(head){
ans++;
head = head->next;
}
return ans;
}
ListNode* removeNthFromEnd(ListNode* head, int n) {
// NOTE01
ListNode* dummy = new ListNode(0,head);
// NOTE02
int size = getLength(head);
// 知道總長度後,就可以知道【順向要刪除的是第幾個節點】
int m = size - n ;
// 重新建立一個新指針,並用這個指針去跑
ListNode* cur = dummy;
for(int i=0;i<m;i++){
cur = cur->next;
}
// 做刪減
cur->next = cur->next->next;
return dummy->next;
}
};