golang力扣leetcode 86.分隔链表

package main
type ListNode struct {
  Val  int
  Next *ListNode
}
func partition(head *ListNode, x int) *ListNode {
  if head == nil {
    return nil
  }
  headDummy := &ListNode{0, head}
  tailDummy := &ListNode{}
  head = headDummy
  tail := tailDummy
  for head.Next != nil {
    if head.Next.Val < x {
      head = head.Next
    } else {
      cnt := head.Next
      head.Next = head.Next.Next
      tail.Next = cnt
      tail = tail.Next
    }
  }
  tail.Next = nil
  head.Next = tailDummy.Next
  return headDummy.Next
}
func main() {
}