Leetcodeleetcode

86. Partition List.go

2019-03-24  本文已影响0人  AnakinSun

新建两个list,然后遍历原来的list,不断的填充

func partition(head *ListNode, x int) *ListNode {

    beforehead := &ListNode{}
    before := beforehead
    afterhead := &ListNode{}
    after := afterhead
    for head != nil {
        if head.Val < x {
            before.Next = head
            before = before.Next
        } else {
            after.Next = head
            after = after.Next
        }
        head = head.Next
    }
    after.Next = nil
    before.Next = afterhead.Next
    return beforehead.Next
}
上一篇 下一篇

猜你喜欢

热点阅读