用Python交换链表中的节点的程序
假设我们有一个列表L和另一个值k。我们必须交换从开头数起的第k个节点和从结尾数起的第k个节点,并在最后返回最终列表。
所以,如果输入是L = [1,5,6,7,1,6,3,9,12],k = 3,那么输出将是[1,5,3,7,1,6,6,9,12],从开头数起的第三个节点是6而从结尾数起的是3,所以它们被交换了。
要解决这个问题,我们将按照以下步骤进行−
- temp := L
- 对于i从0到k-2,做如下操作:
- temp := temp的下一个节点
- firstNode := temp
- secondNode := L
- 当temp的下一个节点不为空时,做如下操作:
- secondNode := secondNode的下一个节点
- temp := temp的下一个节点
- 交换firstNode和secondNode的值
- 返回L
示例
让我们看以下实现以更好地理解。
class ListNode:
def __init__(self, data, next = None):
self.val = data
self.next = next
def make_list(elements):
head = ListNode(elements[0])
for element in elements[1:]:
ptr = head
while ptr.next:
ptr = ptr.next
ptr.next = ListNode(element)
return head
def print_list(head):
ptr = head
print('[', end = "")
while ptr:
print(ptr.val, end = ", ")
ptr = ptr.next
print(']')
def solve(L, k):
temp = L
for i in range(k-1):
temp = temp.next
firstNode = temp
secondNode = L
while temp.next:
secondNode = secondNode.next
temp = temp.next
firstNode.val , secondNode.val = secondNode.val , firstNode.val
return L
L = [1,5,6,7,1,6,3,9,12]
k = 3
print_list(solve(make_list(L), k))
输入
[1,5,6,7,1,6,3,9,12], 3
输出
[1, 5, 3, 7, 1, 6, 6, 9, 12, ]