给你一个链表,删除链表的倒数第 n
个结点,并且返回链表的头结点。
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode() : val(0), next(nullptr) {}
* ListNode(int x) : val(x), next(nullptr) {}
* ListNode(int x, ListNode *next) : val(x), next(next) {}
* };
*/
示例 1:
输入:head = [1,2,3,4,5], n = 2 输出:[1,2,3,5]
示例 2:
输入:head = [1], n = 1 输出:[]
思路
一次遍历?如果我们进行一次遍历同时利用vector存储整张链表的每个节点的地址,
然后访问修改对应位置即可。
解题过程
申请一个新的头结点,与原始链表相连,防止空链表数据的输出异常。
构造vector<ListNode *>address(1,ans),它当前长度为1,存入了我们新申请的头结点。如果head!=nullptr,则将其压入vector,然后递入下一个节点。
遍历结束后,访问address中下标为address.size()-n-1的位置,他是第n个结点的上一个节点(注意address的长度比原链表长度大1个节点,address[0]是我们申请的新头结点),将address.size()-n位置的的next指针与之连接,即抛弃了address.size()-n这个节点。
实测击败全世界。
复杂度
时间复杂度: O(n)
空间复杂度: O(n)
Code
class Solution {
public:
ListNode* removeNthFromEnd(ListNode* head, int n) {
ListNode* ans=new ListNode(0,head);
vector<ListNode*>address(1,ans);
while(head){
address.push_back(head);
head=head->next;
}
address[address.size()-n-1]->next=address[address.size()-n]->next;
return ans->next;
}
};