力扣合并k个有序链表python3实现中的问题

python优先队列：

from dataclasses import dataclass, field
from typing import Any
@dataclass(order=True)
class PrioritizedItem:
    priority: int
    item: Any=field(compare=False)

(priority_number, data)为加入队列的数据形式，其中data应当是可比较类型，如果数据不可比较，则可以忽略data item只比较priority number。
下面直接看和优先队列有关的compare设置方法。

from dataclasses import dataclass, field
from queue import PriorityQueue

@dataclass(order=True)
class Item:
    num1: int
    num2: int
    str_: str

item1 = Item(5, 3, 'c')
item2 = Item(3, 4, 'b')
q = PriorityQueue()
q.put()
q.put()

在上述例子中，如果不作任何改动，则我们在调用get方法的时候，会按照(priority_number, data)中data元组的顺序依次进行比较。比如在上述例子中，按照(num1,num2,str_)依次比较。

>>>q,get()
Item(num1=3, num2=4, str_='b')

而在力扣这个题目中，官方解答的队列入队元素是(l.val,l)也就是第二个元素是个链表，显然链表类型无法比较大小，因此这里我们需要将这里的比较方法做一个修改。

from dataclasses import dataclass, field
from queue import PriorityQueue

@dataclass(order=True)
class Item:
    num1: int
    num2: int=field(compare=False)#即这个类字段不参与比较
    str_: str

item1 = Item(3, 5, 'a')
item2 = Item(3, 4, 'b')
q = PriorityQueue()
q.put(item1)
q.put(item2)

经过上述修改后，在调用get方法时，跳过num2，比较完num1之后直接比较str。

>>>q.get()
Item(num1=3, num2=5, str_='a')

则我们可将力扣官方代码修改如下：

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, val=0, next=None):
#         self.val = val
#         self.next = next
from dataclasses import dataclass, field
from typing import Any
from queue import PriorityQueue
@dataclass(order=True)
class PriorityItem:
    priority_number: int
    item: Any=field(compare=False)
class Solution:
    def mergeKLists(self, lists: List[ListNode]) -> ListNode:
        q = PriorityQueue()
        head = point = ListNode(0)
        for l in lists:
            if l:
                q.put(PriorityItem(l.val, l))
        while not q.empty():
            tmp = q.get()
            val, node = tmp.priority_number, tmp.item
            point.next = ListNode(val)
            point = point.next
            node = node.next
            if node:
                q.put(PriorityItem(node.val, node))
        return head.next

附上力扣官方链接：力扣23合并k个有序链表