단일 연결 목록이 있다고 가정하고 k번째 마지막 노드(0-인덱싱됨)의 값을 찾는지 확인해야 합니다. 우리는 이것을 한 번에 해결해야 합니다.
따라서 입력이 node =[5,4,6,3,4,7], k =2와 같으면 두 번째 마지막(인덱스 3) 노드의 값이 3이므로 출력은 3이 됩니다.
이 문제를 해결하기 위해 다음 단계를 따릅니다. −
-
klast :=노드
-
마지막 :=노드
-
범위 0에서 k에 있는 i에 대해 수행
-
마지막 :=마지막의 다음
-
-
마지막의 다음이 null이 아닌 동안 수행
-
마지막 :=마지막의 다음
-
klast :=klast의 다음
-
-
klast의 반환 값
이해를 돕기 위해 다음 구현을 살펴보겠습니다. −
예
class ListNode: def __init__(self, data, next = None): self.val = data self.next = next def make_list(elements): head = ListNode(elements[0]) for element in elements[1:]: ptr = head while ptr.next: ptr = ptr.next ptr.next = ListNode(element) return head class Solution: def solve(self, node, k): klast = node last = node for i in range(k): last = last.next while last.next: last = last.next klast = klast.next return klast.val ob = Solution() l1 = make_list([5,4,6,3,4,7]) print(ob.solve(l1, 2))
입력
[5,4,6,3,4,7], 2
출력
3