Python判断两个链表是否相等|极客笔记

Python判断两个链表是否相等

在Python中，可以使用一个简单的函数来判断两个链表是否相等。这个函数会逐个比较链表中的每个元素，如果它们的值相同且位置也相同，则认为这两个链表是相等的。

相等链表的定义

在判断链表是否相等之前，我们先来定义一下什么是相等的链表。在这里，我们定义当且仅当链表的长度相同且每个节点的值也相同的情况下，这两个链表才被认为是相等的。

实现Python函数

下面是一个简单的Python函数，用于判断两个链表是否相等：

class ListNode:
    def __init__(self, val=0, next=None):
        self.val = val
        self.next = next

def are_lists_equal(head1, head2):
    curr1, curr2 = head1, head2
    while curr1 and curr2:
        if curr1.val != curr2.val:
            return False
        curr1, curr2 = curr1.next, curr2.next
    return curr1 is None and curr2 is None

在上面的代码中，我们首先定义了一个ListNode类，用于表示链表中的每个节点。然后我们定义了一个are_lists_equal函数，接受两个链表的头节点作为参数，判断它们是否相等。函数通过逐个比较链表中的每个节点的值来实现判断。

测试代码

接下来，我们来测试这个函数是否能正确判断两个链表是否相等。下面是一个简单的测试代码：

# 创建链表1: 1 -> 2 -> 3
head1 = ListNode(1)
head1.next = ListNode(2)
head1.next.next = ListNode(3)

# 创建链表2: 1 -> 2 -> 3
head2 = ListNode(1)
head2.next = ListNode(2)
head2.next.next = ListNode(3)

# 判断链表1和链表2是否相等
print(are_lists_equal(head1, head2))  # 输出: True

在上面的测试代码中，我们首先创建了两个链表head1和head2，它们的内容完全相同。然后我们调用are_lists_equal函数来判断这两个链表是否相等。最后输出为True，说明函数成功判断这两个链表是相等的。