code kata 230827

code_kata_230826/reorder_list_test.go

@@ -42,6 +42,7 @@ func reorderList(head *node) {
 func TestReorderList(t *testing.T) {
 	printList(makeNode(10))
 	reorderList(makeNode(10))
+	reorderList2(makeNode(21))
 }
 
 func printList(head *node) {
@@ -52,6 +53,36 @@ func printList(head *node) {
 	fmt.Println("nil")
 }
 
+func reorderList2(head *node) {
+	printList(head)
+	f, s := head, head
+	for f.next != nil && f.next.next != nil {
+		f = f.next.next
+		s = s.next
+	}
+
+	sh := s.next
+	s.next = nil
+	printList(head)
+	printList(sh)
+	var pre *node
+	for sh != nil {
+		next := sh.next
+		sh.next = pre
+		pre = sh
+		sh = next
+	}
+	printList(pre)
+	p1, p2 := head, pre
+	for p2.next != nil {
+		next1, next2 := p1.next, p2.next
+		p1.next = p2
+		p2.next = next1
+		p1, p2 = next1, next2
+	}
+	printList(head)
+}
+
 func makeNode(count int) *node {
 	var h *node = nil
 	var p *node = nil