[leetcode/python] 206번 Reverse Linked List

https://leetcode.com/problems/reverse-linked-list/description/

Reverse Linked List - LeetCode

 

문제 요약 : 연결 리스트 뒤집기

풀이 로직은 다음과 같다.

- 연결 리스트를 돌면서 result 변수에 거꾸로 담아가기

 

# Definition for singly-linked list.
class ListNode:
    def __init__(self, val=0, next=None):
        self.val = val
        self.next = next
class Solution:
    def reverseList(self, head: Optional[ListNode]) -> Optional[ListNode]:
        result = None
        cur = head

        while cur:
            temp = result
            result = ListNode(cur.val)
            result.next = temp
            cur = cur.next
        
        return result

 

• 시간복잡도 : O(n)
• 개선할 점 : 임시 변수 temp를 쓰지 않고, 바로 swap 하면 더 깔끔하겠다