Posted on 2022-12-06 15:55
Uriel 阅读(53)
评论(0) 编辑 收藏 引用 所属分类:
数据结构 、
闲来无事重切Leet Code
给一个单链表,要求重新排序,前一半为链表的奇数位,后一半为偶数位,e.g.,
Input: head = [1,2,3,4,5]
Output: [1,3,5,2,4]
设置三个指针位置,odd表示当前奇数位处理到哪里,r表示目前已处理到的位置的下一位,even表示当前偶数位处理到哪里,另设置pos记录当前已经处理过多少个(用于判断奇偶)
1.如果当前处理偶数位,odd指向r,r指向next,even不变,e.g.,
1->3->2->4->5->6->7 (此时odd指向3,even指向2,r指向4)
在处理节点4时,指针更新为:odd依旧指向3,even指向4,r指向5,链表顺序不变
2.如果当前处理奇数位,先用临时变量记录第一个偶数位(first_even = odd.next),r的下一位(r_next = r.next),然后分五步改变链表指针顺序
1->3->2->4->5->6->7 (此时odd指向3,even指向4,r指向5)
s1:更新odd.next = r (让3的下一位指向5)
s2:更新r.next = first_even (让5的下一位指向2)
s3:更新even.next = r_next (让4的下一位指向6)
s4:更新odd = r (此时odd指向5)
s5:更新r = r_next (此时r指向6)
更新后链表顺序为:1->3->5->2->4->6->7 (此时odd指向5,even指向4,r指向6)
复杂度O(n),内存O(1)
1 #328
2 #Runtime: 73 ms
3 #Memory: 17.1 MB
4
5 # Definition for singly-linked list.
6 # class ListNode(object):
7 # def __init__(self, val=0, next=None):
8 # self.val = val
9 # self.next = next
10 class Solution(object):
11 def oddEvenList(self, head):
12 """
13 :type head: ListNode
14 :rtype: ListNode
15 """
16 if not head:
17 return head
18 pos = 1
19 r = head.next
20 odd = head
21 even = head
22 while r:
23 if pos % 2:
24 even = r
25 r = r.next
26 else:
27 first_even = odd.next
28 r_next = r.next
29 odd.next = r
30 r.next = first_even
31 even.next = r_next
32 odd = r
33 r = r_next
34 pos += 1
35 return head