【题目】
给定链表的头节点head,实现删除链表的中间节点的函数。
例如:
1,不删除任何节点
1 -> 2,删除节点1
1 -> 2 -> 3,删除节点2
1 -> 2 -> 3 -> 4,删除节点2
1 -> 2 -> 3 -> 4 -> 5,删除节点3
进阶问题:
给定链表的头节点head,整数a和整数b,实现删除位于a/b处节点的函数。
例如:
链表:1 -> 2 -> 3 -> 4 -> 5k,假设a/b的值为r。
如果r等于0,不删除任何节点
如果r在区间(0, 1/5]上,删除节点1
如果r在区间(1/5, 2/5]上,删除节点2
如果r在区间(2/5, 3/5]上,删除节点3
如果r在区间(3/5, 4/5]上,删除节点4
如果r在区间(4/5, 1]上,删除节点5
如果r大于1,不删除任何节点
【基本思路】
原问题。
观察题目,我们可以发现,链表长度每增加2,要删除的节点就往后移动一位。我们只要利用这个规律找到要删除节点的前一个节点,问题就解决了。
进阶问题。
首先,根据链表的长度n,以及a与b的值确定要删除的节点是哪一个节点。计算方法如下:
r = math.ceil(a / b * n),其中ceil函数是向上取整,a / b的值是浮点型。
知道要删除哪一个节点后,只需要找到该节点的前一个节点即可。
【代码实现】
#python3.5
def removeMidNode(head):
if head == None or head.next == None:
return head
if head.next.next == None:
return head.next
pre = head
cur = head.next.next
while cur.next != None and cur.next.next != None:
pre = pre.next
cur = cur.next.next
pre.next = pre.next.next
return head
def removeByRatio(head, a, b):
if head == None or a < 1 or a > b:
return head
n = 0
cur = head
while cur != None:
cur = cur.next
n += 1
n = math.ceil(a / b * n)
if n == 1:
return head.next
cur = head
while n-1 != 1:
cur = cur.next
n -= 1
cur.next = cur.next.next
return head