[LeetCode] 23. Merge k 정렬 리스트 문제 풀이 보고서 (Python)
제목 주소:https://leetcode.com/problems/merge-k-sorted-lists/description/
제목 설명:
Merge k sorted linked lists and return it as one sorted list. Analyze and describe its complexity.
Example:
Input:
[
1->4->5,
1->3->4,
2->6
]
Output: 1->1->2->3->4->4->5->6
제목 의 대의
하나의 링크 에 있 는 k 개의 질서 있 는 링크 를 질서 있 는 링크 로 합 쳤 다.
문제 풀이 방법
방법 1: 매번 최소 값 옮 겨 다 니 기 (TLE)
이 문 제 는 21. Merge Two Sorted Lists 의 확장 으로 고전 문제 에 속 합 니 다. 생각 하기 쉬 운 방법 은 lists 에서 가장 작은 값 을 찾 은 다음 에 현재 의 링크 끝 에 연결 하 는 것 입 니 다.주의해 야 할 것 은 링크 지침 을 수정 하 는 방법 으로 lists 의 머리 요 소 를 업데이트 할 수 없 기 때문에 lists 를 강제로 업데이트 할 수 밖 에 없습니다.
시간 카드 는 링크 의 최소 요 소 를 찾 을 때마다 시간 을 초과 합 니 다.
시간 복잡 도 는 O (N * K), 공간 복잡 도 는 O (1) 다.N 은 결과 링크 의 길이 이 고 K 는 매번 문제 가 제시 하 는 링크 갯 수 입 니 다.
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, x):
# self.val = x
# self.next = None
class Solution:
def mergeKLists(self, lists):
"""
:type lists: List[ListNode]
:rtype: ListNode
"""
head = ListNode(-1)
move = head
while True:
curHead = ListNode(float('inf'))
curIndex = -1
for i, llist in enumerate(lists):
if llist and llist.val < curHead.val:
curHead = llist
curIndex = i
if curHead.val == float('inf'):
break
curNext = curHead.next
move.next = curHead
curHead.next = None
move = curHead
curHead = curNext
lists[curIndex] = curHead
return head.next
방법 2: 작은 루트 저장 값 과 색인
이 방법 은 위 에서 시간 을 초과 한 상황 에 따라 자 연 스 럽 게 생각해 낸 것 이다. 우 리 는 매번 K 개의 체인 헤드 노드 의 최소 치 를 사용 해 야 하기 때문에 모든 링크 의 머리 노드 를 작은 뿌리 더미 안에 넣는다.이렇게 매번 튀 어 나 오 는 것 은 최소 링크 의 값 입 니 다. 그리고 이 값 의 색인 에 따라 Lists 에서 해당 하 는 노드 를 찾 아 끝까지 연결 하면 됩 니 다.
누군가가 사용 하 는 팝 업 더미 에서 가장 작은 값 을 사용 한 다음 에 새로운 노드 를 다시 만 드 는 방식 은 좋 지 않다.
또한 코드 에서 주의해 야 할 문 제 는 방법 과 마찬가지 로 링크 의 머리 결 점 을 업데이트 해 야 한 다 는 것 입 니 다. 지침 을 수정 하 는 방식 으로 수정 할 수 없고 직접 값 을 부여 하여 업데이트 해 야 합 니 다.이 절차 가 적 으 면 색인 에 따라 찾 으 면 오래된 노드 를 계속 얻 을 수 있 습 니 다.
시간 복잡 도 는 O (N) 이 고 공간 복잡 도 는 O (1) 이다.N 은 결과 링크 의 길이 이 고 K 는 매번 문제 가 제시 하 는 링크 갯 수 입 니 다.
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, x):
# self.val = x
# self.next = None
class Solution:
def mergeKLists(self, lists):
"""
:type lists: List[ListNode]
:rtype: ListNode
"""
head = ListNode(-1)
move = head
heap = []
heapq.heapify(heap)
[heapq.heappush(heap, (l.val, i)) for i, l in enumerate(lists) if l]
while heap:
curVal, curIndex = heapq.heappop(heap)
curHead = lists[curIndex]
curNext = curHead.next
move.next = curHead
curHead.next = None
move = curHead
curHead = curNext
if curHead:
lists[curIndex] = curHead
heapq.heappush(heap, (curHead.val, curIndex))
return head.next
방법 3: 작은 루트 저장 값 과 노드
Python 3 에 대해 서 는 노드 를 직접 쌓 아 둘 수 없습니다. 비교 할 수 없 는 문제 가 발생 할 수 있 기 때 문 입 니 다.그러나 python 2 에 대해 서 는 이렇게 할 수 있 기 때문에 노드 의 값 과 노드 를 더미 에 직접 저장 할 수 있 습 니 다. 그러면 매번 튀 어 나 오 는 것 이 가장 작은 값 의 노드 이 고 직접 사용 하면 됩 니 다.
# Definition for singly-linked list.
# class ListNode(object):
# def __init__(self, x):
# self.val = x
# self.next = None
class Solution(object):
def mergeKLists(self, lists):
"""
:type lists: List[ListNode]
:rtype: ListNode
"""
head = ListNode(-1)
move = head
heap = []
heapq.heapify(heap)
[heapq.heappush(heap, (l.val, l)) for i, l in enumerate(lists) if l]
while heap:
curVal, curHead = heapq.heappop(heap)
curNext = curHead.next
move.next = curHead
curHead.next = None
move = curHead
curHead = curNext
if curHead:
heapq.heappush(heap, (curHead.val, curHead))
return head.next
시간 복잡 도 는 O (N) 이 고 공간 복잡 도 는 O (1) 이다.N 은 결과 링크 의 길이 이 고 K 는 매번 문제 가 제시 하 는 링크 갯 수 입 니 다.
참고 자료:
날짜.
2018 년 10 월 16 일 - 비 오 는 날 은 그래도 편 해 요.
이 내용에 흥미가 있습니까?
현재 기사가 여러분의 문제를 해결하지 못하는 경우 AI 엔진은 머신러닝 분석(스마트 모델이 방금 만들어져 부정확한 경우가 있을 수 있음)을 통해 가장 유사한 기사를 추천합니다:
python 문자열 입력으로 모든 유효한 IP 주소 생성(LeetCode 93번 문제)이 문제의 공식 난이도는 Medium으로 좋아요 1296, 반대 505, 통과율 35.4%를 눌렀다.각 항목의 지표로 말하자면 보기에는 약간 규범에 맞는 것 같지만, 실제로도 확실히 그렇다.이 문제의 해법과 의도는 ...
텍스트를 자유롭게 공유하거나 복사할 수 있습니다.하지만 이 문서의 URL은 참조 URL로 남겨 두십시오.
CC BY-SA 2.5, CC BY-SA 3.0 및 CC BY-SA 4.0에 따라 라이센스가 부여됩니다.