leetcode19. Remove Nth Node From End of List刪除連結串列的倒數第N個節點
阿新 • • 發佈:2018-12-22
題目:
Given a linked list, remove the n-th node from the end of list and return its head. Example: Given linked list: 1->2->3->4->5, and n = 2. After removing the second node from the end, the linked list becomes 1->2->3->5. Note: Given n will always be valid. |
給定一個連結串列,刪除連結串列的倒數第 n 個節點,並且返回連結串列的頭結點。 示例: 給定一個連結串列: 1->2->3->4->5, 和 n = 2. 當刪除了倒數第二個節點後,連結串列變為 1->2->3->5. 說明: 給定的 n 保證是有效的。 |
思路:兩個指標,cur先移動n步,pre指向第一個節點。然後兩個一起移動,直到cur移動到最後一個元素,pre所指向的就是倒數第n個節點。注意的是連結串列長度為n刪除倒數第n個幾點, 那就直接返回head->next;
/** * Definition for singly-linked list. * struct ListNode { * int val; * ListNode *next; * ListNode(int x) : val(x), next(NULL) {} * }; */ class Solution { public: ListNode* removeNthFromEnd(ListNode* head, int n) { if(head==NULL) return NULL; ListNode *cur=head,*pre=head; for(int i=0;i<n;++i) cur=cur->next; if(cur==NULL) return head->next; while(cur->next) { cur=cur->next; pre=pre->next; } pre->next=pre->next->next; return head; } };