86. 分隔链表 golang

86. 分隔链表

只把小的放在一边，大于的不用放到右边（这个题的题目比较晦涩，很多人说看不懂题，我也没看懂，解法也没什么特殊，半天没读懂题，看了别人的解法）

给定一个链表和一个特定值 x，对链表进行分隔，使得所有小于 x 的节点都在大于或等于 x 的节点之前。

你应当保留两个分区中每个节点的初始相对位置。

示例:

输入: head = 1->4->3->2->5->2, x = 3
输出: 1->2->2->4->3->5

Code

/**
 * Definition for singly-linked list.
 * type ListNode struct {
 *     Val int
 *     Next *ListNode
 * }
 */
func partition(head *ListNode, x int) *ListNode {
    if head == nil || head.Next == nil{
        return head
    }
    less:= new(ListNode)
    cur_l := less
    more := new(ListNode)
    cur_m :=more
    cur := head

    for cur !=nil{
        if cur.Val < x {
            cur_l.Next = &ListNode{cur.Val,nil}
            cur_l = cur_l.Next
        }else{
            cur_m.Next = &ListNode{cur.Val,nil}
            cur_m = cur_m.Next
        }
        cur = cur.Next
    }
    
    cur_l.Next = more.Next
    return less.Next
}