力扣算法练习BM9——删除链表的倒数第n个节点

题目

给定一个链表，删除链表的倒数第 n 个节点并返回链表的头指针
例如，

给出的链表为: 1\to 2\to 3\to 4\to 51→2→3→4→5, n= 2n=2.
删除了链表的倒数第 nn 个节点之后,链表变为1\to 2\to 3\to 51→2→3→5.

数据范围： 链表长度 0\le n \le 10000≤n≤1000，链表中任意节点的值满足 0 \le val \le 1000≤val≤100

要求：空间复杂度 O(1)O(1)，时间复杂度 O(n)O(n)
备注：

题目保证 nn 一定是有效的

示例1

输入：{1,2},2

复制返回值：{2}

解题思路

1.遍历链表，获取链表的长度
2.若n为1，链表长度(k)为1返回空；k不为1，返回删除最后一个节点的链表；
3.若删除头节点，则返回头节点之后的链表；若不是删除头节点，则直接返回删除节点之后的链表

题解

# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None
#
# 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
#
# 
# @param head ListNode类 
# @param n int整型 
# @return ListNode类
#
class Solution:
    def removeNthFromEnd(self , head: ListNode, n: int) -> ListNode: 
        phead = head
        p = phead
        k = 1
        # 1.获取链表个数
        while p.next:
            p = p.next
            k += 1
        p = phead
        # 2.若n为1，链表长度(k)为1返回空,k不为1，返回删除最后一个节点的链表；
        if n == 1:
            if k ==1: return None       
            for i in range(k-2):
                p = p.next 
            p.next = None
            return phead
        #若删除头节点，则返回头节点之后的链表；若不是删除头节点，则返回删除节点之后的链表
        elif n == k: return phead.next
        else:
            for i in range(k):
                if i==k-n-1:
                    p.next = p.next.next
                    continue
                p = p.next
        return phead