leetcode86 分割链表

谜底6662023-09-20 10:54

题目

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

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

示例

输入：head = $1,4,3,2,5,2$ , x = 3

输出： $1,2,2,4,3,5$

解析

首先看这道题的时候，那个示例半天没看明白，它要的是比较之后还是相对序列，并不是整体都排序了。然后直接开两个链表来做就可以

go 复制代码

/**
 * Definition for singly-linked list.
 * type ListNode struct {
 *     Val int
 *     Next *ListNode
 * }
 */
func partition(head *ListNode, x int) *ListNode {
	small := &ListNode{}
	large := &ListNode{} // 定义两个新链表
	smallHead := small
	largeHead := large // 定义两个头节点
	for head != nil {
		if head.Val < x {
			small.Next = head
			small = small.Next
		} else {
			large.Next = head
			large = large.Next
		}
		head = head.Next
	}
	large.Next = nil 
	small.Next = largeHead.Next
	return smallHead.Next
}