删除链表的倒数第 N 个结点

删除链表的倒数第 N 个结点

题目链接

删除倒数第n个结点

题目

给你一个链表，删除链表的倒数第 n 个结点，并且返回链表的头结点
链表中结点的数目为 sz
1 <= sz <= 30
0 <= Node.val <= 100
1 <= n <= sz

输入输出示例

输入：head = [1,2,3,4,5], n = 2
输出：[1,2,3,5]
示例 2：

输入：head = [1], n = 1
输出：[]
示例 3：

思路

首先第一个想到的就是先遍历一遍统计链表结点个数，然后再遍历一遍找到倒数第n个结点进行删除

参考代码1

struct ListNode* removeNthFromEnd(struct ListNode* head, int n){
    struct ListNode *p,*q;
	p = (struct ListNode*)malloc(sizeof(struct ListNode));
	p = head;
	int count1=0,count2=0;
	while(p!=NULL)//遍历一遍统计链表结点个数
	{
		p = p->next;
		count1++;
	}
    if(count1 == 1)//若链表只有一个结点那么返回的一定是空链表
    {
        head = NULL;
        return head;
    }
    else if(count1 == n)//删除头结点
    {
        head->val = head->next->val;
        head->next = head->next->next;
        return head;
    }
	p = head;
	q = head;
	while(count2!=count1-n)
	{
		count2++;
        q = p;
		p = p->next;
	}
    q->next = p->next;
    free(p);
	return head;
}

或者采用双指针，不需要循环两遍，p比q超前n

参考代码2

struct ListNode* removeNthFromEnd(struct ListNode* head, int n){
    struct ListNode *p=head,*q=head;
    if(head->next==NULL) return NULL;//头结点的下一个为空说明该链表只有一个元素，删除后就为空链表
    while(p!=NULL)
    {
        if(n!=-1) n--;
        else q = q->next;
        p = p->next;
    }
    if(n==0)//n=0说明要删除的是头结点
    {
        head->val = head->next->val;
        head->next = head->next->next;
    }
    else q->next = q->next->next;//n=-1时就说明要删除的是q的下一个结点
	return head;
}