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
}