刷題--程序員面試金典--面試題 02.04. 分割鏈表

面試題 02.04. 分割鏈表

編寫程序以 x 爲基準分割鏈表，使得所有小於 x 的節點排在大於或等於 x 的節點之前。如果鏈表中包含 x，x 只需出現在小於 x 的元素之後(如下所示)。分割元素 x 只需處於“右半部分”即可，其不需要被置於左右兩部分之間。

示例:

輸入: head = 3->5->8->5->10->2->1, x = 5
輸出: 3->1->2->10->5->5->8

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/partition-list-lcci
著作權歸領釦網絡所有。商業轉載請聯繫官方授權，非商業轉載請註明出處。

---

func partition(head *ListNode, x int) *ListNode {
    lHead := &ListNode{0,nil}
    rHead := &ListNode{0,nil}

    l := lHead
    r := rHead

    for head != nil {
        t := &ListNode{head.Val,nil}
        if head.Val < x {
            l.Next = t
            l = l.Next
        } else {
            r.Next = t
            r = r.Next
        }
        head = head.Next
    }

    l.Next = rHead.Next

    return lHead.Next
}