목록 L과 다른 값 k가 있다고 가정합니다. 시작에서 k번째 노드와 끝에서 k번째 노드를 교환하고 마지막에 최종 목록을 반환해야 합니다.
따라서 입력이 L =[1,5,6,7,1,6,3,9,12] k =3과 같으면 출력은 [1,5,3,7,1,6, 6,9,12], 시작부터 3번째 노드는 6, 끝 노드는 3이므로 스왑됩니다.
이 문제를 해결하기 위해 다음 단계를 따릅니다. −
- 온도 :=L
- 0~k-2 범위의 i에 대해 다음을 수행합니다.
- temp :=다음 temp
- firstNode :=임시
- secondNode :=L
- temp의 다음이 null이 아닌 동안 do
- secondNode :=secondNode의 다음
- temp :=다음 temp
- firstNode의 값과 secondNode의 값 교환
- 반환 L
예시
이해를 돕기 위해 다음 구현을 살펴보겠습니다. −
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
def print_list(head):
ptr = head
print('[', end = "")
while ptr:
print(ptr.val, end = ", ")
ptr = ptr.next
print(']')
def solve(L, k):
temp = L
for i in range(k-1):
temp = temp.next
firstNode = temp
secondNode = L
while temp.next:
secondNode = secondNode.next
temp = temp.next
firstNode.val , secondNode.val = secondNode.val , firstNode.val
return L
L = [1,5,6,7,1,6,3,9,12]
k = 3
print_list(solve(make_list(L), k)) 입력
[1,5,6,7,1,6,3,9,12], 3
출력
[1, 5, 3, 7, 1, 6, 6, 9, 12, ]