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个有序链表