연결된 목록이 있다고 가정합니다. 연결 리스트의 전반부를 가져와서 후반부를 접은 다음 교차하는 노드의 합을 취해 병합해야 합니다. 마지막으로 연결 목록의 결과 헤드를 반환해야 합니다.
따라서 입력이 [5,8,1,2,4,7,5]와 같으면 출력은 [2, 5, 15, 10, ]
가 됩니다.이 문제를 해결하기 위해 다음 단계를 따릅니다. −
- 온도 :=0
- ptr :=노드
- ptr이 null이 아닌 동안 do
- temp :=온도 + 1
- ptr :=ptr의 다음
- t :=온도의 몫 / 2
- m :=노드
- stk :=새 스택
- t가 0이 아닌 동안 do
- m의 값을 stk로 푸시
- tmp :=다음 m
- m 옆:=null
- m :=tmp
- t :=t - 1
- 노드:=m
- temp가 짝수이면
- m :=m의 다음
- m이 null이 아닌 동안 do
- m의 값 :=m의 값 + 스택 맨 위 요소.
- stk에서 팝
- m :=m의 다음
- 반환 노드
이해를 돕기 위해 다음 구현을 살펴보겠습니다. −
예시
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(']')
class Solution:
def solve(self, node):
temp = 0
ptr = node
while ptr:
temp += 1
ptr = ptr.next
t = temp // 2
m = node
stk = []
while t:
stk.append(m.val)
tmp = m.next
m.next = None
m = tmp
t -= 1
node = m
if temp % 2 != 0:
m = m.next
while m:
m.val += stk.pop()
m = m.next
return node
ob = Solution()
head = make_list([5,8,1,2,4,7,5])
print_list(ob.solve(head)) 입력
[5,8,1,2,4,7,5]
출력
[2, 5, 15, 10, ]