《leetcode-go》划分链表

最新推荐文章于 2022-08-14 11:51:23 发布

原创最新推荐文章于 2022-08-14 11:51:23 发布 · 183 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#链表 #leetcode #golang

leetCode-go 专栏收录该内容

81 篇文章

订阅专栏

该博客讨论了一个算法问题，涉及对单链表进行操作，将所有值小于x的节点移到值大于等于x的节点之前，同时保持原有顺序。解决方案是创建两个新的链表，分别用于存储小于x和大于等于x的节点，然后将它们连接起来。此算法的时间复杂度为O(n)，空间复杂度为O(1)。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

给出一个长度为 n 的单链表和一个值 x ，请返回一个链表的头结点，要求新链表中小于 x 的节点全部在大于等于 x 的节点左侧，并且两个部分之内的节点之间与原来的链表要保持相对顺序不变。

复杂度要求：

时间 O（n）

空间 O（1）

思路：新建一个比x小的部分的头，再建一个比x大的部分的头，然后遍历把他们分别放到小的部分和大的部分。然后两个部分连起来。返回小的头的next。

package main
import . "nc_tools"
/*
 * type ListNode struct{
 *   Val int
 *   Next *ListNode
 * }
 */

/**
  * 
  * @param head ListNode类 
  * @param x int整型 
  * @return ListNode类
*/
func partition( head *ListNode ,  x int ) *ListNode {
    // write code here
    if head == nil {
        return head
    }
    less := new(ListNode)
	newHead := less
	large := new(ListNode)
	newLarge := large
	for head != nil {
		if head.Val < x {
			less.Next = head
			less = less.Next
		} else {
			large.Next = head
			large = large.Next
		}
		head = head.Next
        less.Next = nil
        large.Next = nil
	}
	less.Next = newLarge.Next
	return newHead.Next
}