最直接的想法肯定是遍历一遍求出长度,然后用这个这个长度减去n-1定位到要删除的节点,然后再次遍历用一个变量进行计数,然后删除。
这是最低效的办法,提高算法效率的一个基本思想就是:在一次遍历中完成多个任务,这需要在设计算法时抓住题目要求的本质,剔除那些看完题目直接浮现在脑海但是不必要的步骤,同时在必要的步骤中尽可能多地搭载任务。在实现算法时注意各种可能拖慢程序的实现方法。这些注意点可以在计算两数之和的每次求和计算进位等处理方法中可以看出。
很明显,这里是必须要进行遍历的,但是能不能在一次遍历中就定位到要删除的节点?考虑到遍历结束条件是Node == None,往前n个就是要删除的节点,那么可以用另一个变量记录Node前第n个节点,这样就能一次遍历实现了。
class Solution:
def removeNthFromEnd(self, head: ListNode, n: int) -> ListNode:
CurNode = head
CurList = head
m = 0
while CurNode != None:
m += 1
CurNode = CurNode.next
if m > n:
preCurList = CurList
CurList = CurList.next
if m == n:
return head.next
preCurList.next = CurList.next
return head
这个实现方法很慢,平均76ms,只打败了27%的用户,这里拖慢程序的一个核心点就是循环中那个if m> n:只需要判断出一次m>n,明显后面的每一次循环内部都不需要再判断了,所以该怎么改?
这种在循环中的只需要一次的条件判断,可以处理成两个循环,第一个循环以条件判断为循环结束条件,第二个循环则以原循环退出条件为条件。
class Solution:
def removeNthFromEnd(self, head: ListNode, n: int) -> ListNode:
CurNode = head
CurList = ListNode(0)
CurList.next = head
while n > 0:
n -= 1
CurNode = CurNode.next
if CurNode == None:
return head.next
while CurNode != None:
CurNode = CurNode.next
CurList = CurList.next
CurList.next = CurList.next.next
return head
这次平均60ms,大概打败56%的用户,这里除了上述的优化之外,还有一个小技巧就是pre指针的设置,这里不是记录要删除的节点,而是记录其前驱,这样很明显可以节省一个变量空间以及运行时间。
再优化就是参考别人的题解了
class Solution:
def removeNthFromEnd(self, head: ListNode, n: int) -> ListNode:
CurList = ListNode(0)
CurList.next = head
start = CurList
end = CurList
while n != 0:
n -= 1
start = start.next
while start.next != None:
start = start.next
end = end.next
end.next = end.next.next
return CurList.next
这次优化的主要方面就是:直接去掉了两个循环之间的if,设置了pre除了在提升效率之外,还能防止一些边界情况