给定链表,删除倒数第n个链表节点
输入:链表的头节点,n
输出:链表的头节点
要求:一次提交通过
边界: n = 1 index=size-1
n = size index = 0
思路:最简单的方法为,先遍历一遍,求出来总长度,根据总长度求出需要删除的节点位置,然后遍历到位置,进行删除,返回。
时间复杂度还是O(size)
减小复杂度只能通过不进行第二遍遍历。用vector存储相应位置,进行操作。
ListNode* removeNthFromEnd(ListNode* head, int n) {
vector<ListNode*> nodes;
ListNode* tempNode = head;
while(tempNode != NULL){
nodes.push_back(tempNode);
tempNode = tempNode->next;
}
int index = nodes.size() - n;
if(1 == n&& index == 0) return NULL;
if(index == 0){
delete head;
head = nodes[index+1];
}else if(1 == n){
nodes[index-1]->next = NULL;
delete nodes[index];
}else{
nodes[index-1]->next = nodes[index+1];
delete nodes[index];
}
return head;
}
执行用时 : 16 ms, 在Remove Nth Node From End of List的C++提交中击败了11.35% 的用户
内存消耗 : 9.8 MB, 在Remove Nth Node From End of List的C++提交中击败了0.76% 的用户
按照惯例,学习一下最速大佬:
ListNode* removeNthFromEnd(ListNode* head, int n) {
ListNode* h = new ListNode(0);
h->next = head;
ListNode* p = head;
ListNode* q = h;
while(n--){
p = p->next;
}
while(p){
p = p->next;
q = q->next;
}
q->next = q->next->next;
return h->next;
}
执行用时 : 16 ms, 在Remove Nth Node From End of List的C++提交中击败了11.35% 的用户
内存消耗 : 9.6 MB, 在Remove Nth Node From End of List的C++提交中击败了0.76% 的用户
时间和我一样。说明我的效率还是挺高的,不过可以看一下他的方法。
比较巧妙地是,他首先先建立了一个哨兵节点。把头节点等特殊情况直接不予考虑。用了两个指针,p指针初始位置为n,q初始位置为0,然后p和q一起往后移位。当p到了尾节点,说明q和q同时向右移了size-n个位置,因此q地位置为倒数第n个的前一个。于是进行操作,然后直接返回head。所用空间极少,遍历也一遍。不过我好像也是的。但是这个解法没有删除节点,只是将其丢弃,可能会造成内存泄漏。