如何在main函数中使用
时间: 2024-03-25 19:37:12 浏览: 21
好的,让我来给你一个在 main 函数中使用的示例代码。
假设你已经有一个链表的头结点 head,可以将其传入 deleteDuplication 函数中,然后将返回的链表头结点再赋值给 head。最后,你可以遍历链表打印所有结点的值,以验证算法的正确性。
具体代码如下:
```python
class ListNode:
def __init__(self, val=0, next=None):
self.val = val
self.next = next
class Solution:
def deleteDuplication(self, head: ListNode) -> ListNode:
# 定义 pre 和 cur 指针
pre, cur = None, head
while cur:
# 如果当前结点 cur 是重复结点
if cur.next and cur.val == cur.next.val:
val = cur.val
# 使用 while 循环找到所有重复结点并删除
while cur and cur.val == val:
cur = cur.next
# 如果 pre 为 None,说明头结点也是重复结点,直接将 head 指向 cur
if not pre:
head = cur
# 否则将 pre 的 next 指向 cur
else:
pre.next = cur
# 如果当前结点 cur 不是重复结点
else:
# pre 和 cur 同时向后移动一个结点
pre, cur = cur, cur.next
return head
if __name__ == '__main__':
# 创建链表 1->2->3->3->4->4->5
node1 = ListNode(1)
node2 = ListNode(2)
node3 = ListNode(3)
node4 = ListNode(3)
node5 = ListNode(4)
node6 = ListNode(4)
node7 = ListNode(5)
node1.next = node2
node2.next = node3
node3.next = node4
node4.next = node5
node5.next = node6
node6.next = node7
# 删除重复结点
solution = Solution()
head = solution.deleteDuplication(node1)
# 打印链表所有结点的值
while head:
print(head.val, end=' ')
head = head.next
```
输出结果为:1 2 5
这样,你就可以在 main 函数中使用这个算法并验证其正确性了。
相关推荐
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)